python - 如何将两个嵌套字典合并在一起？

Question

我有两个嵌套字典，它们在第一层有重叠的键，但在第二层有不同的键。我想合并它们，以便新字典包含所有键。

A = {'id1': {'key1': 1, 'key2': 2 }, 'id2':{'key1': 3, 'key2': 4 }}
B = {'id1': {'key3': 5}, 'id2': {'key3': 6}}

结果应该是

A_B = {'id1': {'key1': 1, 'key2': 2, 'key3': 5}, 'id2':{'key1': 3, 'key2': 4, 'key3': 6}}

我知道我可以做一个for循环

for key in A:
    A[key].update(B[key])

但我想知道是否有更清洁的解决方案。

Answer 1

您可以将它们合并到字典理解中:

A_B = {k: dict(A.get(k, {}), **B.get(k, {})) for k in A.viewkeys() | B.viewkeys()}

这使用 Python 2 dictionary keys view object ;在 Python 3 中，使用 dict.keys() 而不是 dict.viewkeys()。

这将合并所有键，即使它们只存在于一个或另一个字典中。

根据您的输入进行演示:

>>> A = {'id1': {'key1': 1, 'key2': 2 }, 'id2':{'key1': 3, 'key2': 4 }}
>>> B = {'id1': {'key3': 5}, 'id2': {'key3': 6}}
>>> {k: dict(A.get(k, {}), **B.get(k, {})) for k in A.viewkeys() | B.viewkeys()}
{'id2': {'key3': 6, 'key2': 4, 'key1': 3}, 'id1': {'key3': 5, 'key2': 2, 'key1': 1}}

或使用更多不同的键输入:

>>> C = {'foo': {'spam': 1, 'ham': 2}, 'bar': {'spam': 43, 'eggs': 81}}
>>> D = {'baz': {'vikings': 'singing'}, 'bar': {'monty': 'python'}}
>>> {k: dict(C.get(k, {}), **D.get(k, {})) for k in C.viewkeys() | D.viewkeys()}
{'bar': {'eggs': 81, 'monty': 'python', 'spam': 43}, 'foo': {'ham': 2, 'spam': 1}, 'baz': {'vikings': 'singing'}}