python - 如何将具有相同键的字典组合起来？

Question

我有一个像这样的字典列表：

dicts = [
    {'key_a': valuex1,
     'key_b': valuex2,
     'key_c': valuex3},

    {'key_a': valuey1,
     'key_b': valuey2,
     'key_c': valuey3},

    {'key_a': valuez1,
     'key_b': valuez2,
     'key_c': valuez3}
]

我想用这些来构建一个像这样的大字典：

big_dict = {
    'key_a': [valuex1, valuey1, valuez1],
    'key_b': [valuex2, valuey2, valuez2],
    'key_c': [valuex3, valuey3, valuez3]
}

我有什么优雅的“拉链”方式吗？

所有的键总是相同的。

Answer 1

big_dict = {}
for k in dicts[0]:
    big_dict[k] = [d[k] for d in dicts]

或者，使用dict 理解：

{k: [d[k] for d in dicts] for k in dicts[0]}

Answer 2

您可以使用collections.defaultdict. 这个解决方案的好处是它不需要键在字典之间保持一致，并且它仍然保持最小的 O( n ) 时间复杂度。

from collections import defaultdict

dict_list = [{'key_a': 'valuex1', 'key_b': 'valuex2', 'key_c': 'valuex3'},
             {'key_a': 'valuey1', 'key_b': 'valuey2', 'key_c': 'valuey3'},
             {'key_a': 'valuez1', 'key_b': 'valuez2', 'key_c': 'valuez3'}]            

d = defaultdict(list)
for myd in dict_list:
    for k, v in myd.items():
        d[k].append(v)

结果：

print(d)

defaultdict(list,
            {'key_a': ['valuex1', 'valuey1', 'valuez1'],
             'key_b': ['valuex2', 'valuey2', 'valuez2'],
             'key_c': ['valuex3', 'valuey3', 'valuez3']})

Answer 3

如果所有 dicts 具有相同的键集，这将起作用：

dict((k, [d[k] for d in dictList]) for k in dictList[0])

如果它们可能有不同的键，您需要首先通过对各种 dicts 的键执行 set unions 来构建一组键：

allKeys = reduce(operator.or_, (set(d.keys()) for d in dictList), set())

然后，您需要防止某些字典中缺少键：

dict((k, [d[k] for d in [a, b] if k in d]) for k in allKeys)

Answer 4

如果您乐于使用第三方库，您可以使用 Pandas。构造pd.DataFrame函数直接接受字典列表：

import pandas as pd

res = pd.DataFrame(dictList).to_dict(orient='list')

{'key_a': ['valuex1', 'valuey1', 'valuez1'],
 'key_b': ['valuex2', 'valuey2', 'valuez2'],
 'key_c': ['valuex3', 'valuey3', 'valuez3']}

Answer 5

您可以通过以下方式合并字典：

def merge_dicts(dict_list, separator=''):
    """
    Merges list of dictionaries to a single dictionary, Concatenates values with the same key.
    :param dict_list: list of dictionaries to be merged.
    :param separator: separator to be inserted between values of same key.
    :return: Merged dictionary.
    """
    return {k1: separator.join([d[k1] for d in dict_list if k1 in d])
            for k1 in set(reduce(lambda x, y: x+y, [k.keys() for k in dict_list]))
    }