如何在Python中合并两个字典列表？

Question

我有2个类似这样的列表：

l1 = [{'a': 1, 'b': 2, 'c': 3, 'd': 4}, {'a': 5, 'b': 6, 'c': 7, 'd': 8}]

l2 = [{'a': 5, 'b': 6, 'e': 100}, {'a': 1, 'b': 2, 'e': 101}]

我希望获得一个名为l3的列表，它是l1和l2的连接，其中'a'和'b'的值在l1和l2中都相等。

l3 = [{'a': 1, 'b: 2, 'c': 3, 'd': 4, 'e': 101}, {'a': 5, 'b: 6, 'c': 7, 'd': 8, 'e': 100}]

我该怎么做？

- Tung

Answer 1

你应该在一个字典中累积结果。你应该使用'a'和'b'的值来形成这个字典的键。

在这里，我使用了一个defaultdict来累积条目。

l1 = [{'a': 1, 'b': 2, 'c': 3, 'd': 4}, {'a': 5, 'b': 6, 'c': 7, 'd': 8}]
l2 = [{'a': 5, 'b': 6, 'e': 100}, {'a': 1, 'b': 2, 'e': 101}]

from collections import defaultdict
D = defaultdict(dict)
for lst in l1, l2:
    for item in lst:
        key = item['a'], item['b']
        D[key].update(item)

l3 = D.values()
print l3

输出：

[{'a': 1, 'c': 3, 'b': 2, 'e': 101, 'd': 4}, {'a': 5, 'c': 7, 'b': 6, 'e': 100, 'd': 8}]

- John La Rooy

Answer 2

简单的列表操作也可以为您完成此操作：

l1 = [{'a': 1, 'b': 2, 'c': 3, 'd': 4}, {'a': 5, 'b': 6, 'c': 7, 'd': 8}]
l2 = [{'a': 5, 'b': 6, 'e': 100}, {'a': 1, 'b': 2, 'e': 101}]
l3 = []

for i in range(len(l1)):
    for j in range(len(l2)):
        if l1[i]['a'] == l2[j]['a'] and l1[i]['b'] == l2[j]['b']:
            l3.append(dict(l1[i]))
            l3[i].update(l2[j])

- Iron Fist

---

你应该使用==而不是is来比较值。 - John La Rooy

@JohnLaRooy，但我记得曾经读过“is”比“==”更快...不是吗？ - Iron Fist

可能是因为is只比较身份(identity)。你之所以能在这里得逞，是因为你正在测试非常小的被interned的int值。==是正确比较这些值的唯一方法。 - John La Rooy

@JohnLaRooy...好的，谢谢你提醒我.. :) - Iron Fist

Answer 3

我的方法是按照键（即键 a + b）对组合列表进行排序。然后，对于具有相似键的每组字典，将它们合并：

from itertools import groupby

def ab_key(dic):
    return dic['a'], dic['b']

def combine_lists_of_dicts(list_of_dic1, list_of_dic2, keyfunc):
    for key, dic_of_same_key in groupby(sorted(list_of_dic1 + list_of_dic2, key=keyfunc), keyfunc):
        combined_dic = {}
        for dic in dic_of_same_key:
            combined_dic.update(dic)
        yield combined_dic

l1 = [{'a': 1, 'b': 2, 'c': 3, 'd': 4}, {'a': 5, 'b': 6, 'c': 7, 'd': 8}]
l2 = [{'a': 5, 'b': 6, 'e': 100}, {'a': 1, 'b': 2, 'e': 101}]

for dic in combine_lists_of_dicts(l1, l2, ab_key):
    print dic

讨论

• ab_key 函数返回键 a 和 b 的值的元组，用于对分组进行排序
• groupby 函数将所有具有相似键的字典分组在一起
• 该解决方案比 John La Rooy 的方案效率低，但对于小列表应该可以正常工作

- Hai Vu

Answer 4

使用pandas可以轻松快捷地实现一个不错的解决方案。

l1 = [{'a': 1, 'b': 2, 'c': 3, 'd': 4}, {'a': 5, 'b': 6, 'c': 7, 'd': 8}]
l2 = [{'a': 5, 'b': 6, 'e': 100}, {'a': 1, 'b': 2, 'e': 101}]

import pandas as pd
pd.DataFrame(l1).merge(pd.DataFrame(l2), on=['a','b']).to_dict('records')

- SimonT