Python：使用字典理解附加到值列表

Question

我有以下场景 - 考虑如下字典列表 (input):

data = [{'key1' : 'k1v1', 'key2': 'k2v1'}, {'key1': 'k1v2', 'key2': 'k2v2'}, 
{'key1': 'k1v3', 'key2': 'k2v3'}, {'key1': 'k1v4', 'key2': 'k2v4'}]

可以看出，字典中的键是相同的。所以我想要做的是以下（期望输出）：

{'key1': ['k1v1', 'k1v2', 'k1v3', 'k1v4'], 'key2': ['k2v1', 'k2v2', 'k2v3', 'k2v4']}

为此，我可以使用以下简单直接的嵌套循环：

data_dict = dict()
for d in data:
    for k, v in d.items():
        if k in data_dict:
            data_dict[k].append(v)
        else:
            data_dict[k] = [v]

问题：我们可以使用字典理解来做到这一点吗？换句话说，我们可以在字典理解中附加到现有值列表吗？

Answer 1

这是我能做到的最好的了：

d = {}
for k in data[0].keys():
    d[k] = list(d[k] for d in ds)

这是一个更接近字典理解的解决方案

data_dict = {
    k: [d.get(k) for d in data]
    for k in set().union(*data)
}

以下语法生成的代码具有最快的时序：

data_dict = {k: [dt[k] for dt in data] for k in data[0].keys()}

# Timing:
# 1.08 µs ± 18.2 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

Python：使用字典理解附加到值列表

Python: append to value-list using dictionary-comprehension

dictionary

python-3.x

dictionary-comprehension