如何使用相同的主键组合两个嵌套字典

Question

我有两个具有相同主键的嵌套字典：

dict1 = {'person1': {'name': 'John', 'sex': 'Male'},
         'person2': {'name': 'Marie', 'sex': 'Female'},
         'person3': {'name': 'Luna', 'sex': 'Female'},
         'person4': {'name': 'Peter', 'sex': 'Male'}}

dict2 = {'person1': {'weight': '81.1', 'age': '27'},
         'person2': {'weight': '56.7', 'age': '22'},
         'person3': {'weight': '63.4', 'age': '24'},
         'person4': {'weight': '79.1', 'age': '29'}}

所以我想通过 dict2 中的键值对来丰富 dict 1。

我可以用 for 循环来做到这一点...

for key in dict2:
        dict2[key]['age'] = dict1[key]['age']
        dict2[key]['weight'] = dict2[key]['weight']

结果：

dict2 = {'person1': {'name': 'John', 'sex': 'Male', 'weight': '81.1', 'age': '27'},
         'person2': {'name': 'Marie', 'sex': 'Female', 'weight': '56.7', 'age': '22'},
         'person3': {'name': 'Luna', 'sex': 'Female', 'weight': '63.4', 'age': '24'},
         'person4': {'name': 'Peter', 'sex': 'Male', 'weight': '79.1', 'age': '29'}}

...但是有没有一种更 Pythonic 的方法来做到这一点 - 例如使用 dict 理解？

Answer 1

是的：

dict3 = {k: {**v, **dict2[k]} for k, v in dict1.items()}

首先，使用 .items() 同时迭代键和值。

然后，对于每个键 k，您希望该值是通过转储（或解构）v 和 dict2[k]在其中。

Python 更新 >= 3.9：

感谢 @mwo 提到管道 | 操作数：

dict3 = {k: v | dict2[k] for k, v in dict1.items()}

Answer 2

如果您可以控制数据源，请展平字典，然后使用 update 方法。例如：

dict1 = {('person1', 'name'): 'John'}
dict2 = {('person1', 'weight'): 81.1}
dict1.update(dict2)

>>> dict1
{('person1', 'name'): 'John', 
 ('person1', 'weight'): 81.1}

处理这种数据结构要容易得多，但如果您受困于嵌套字典，您可以使用 NestedDict 通过类似的接口实现相同的结果。

from ndicts import NestedDict

nd1 = NestedDict(dict1) 
nd2 = NestedDict(dict2) 
nd1.update(nd2)

>>> nd1
NestedDict(
    {'person1': {'name': 'John', 'weight': 81.1}}
)

如果您需要结果作为字典，请使用 nd1.to_dict() 。

安装ndicts pip install ndicts。

Answer 3

答案不考虑要合并的字典中深度超过 2 的级别，
要合并到任意深度，请使用以下递归：

def merge_nested_dicts(dict1, dict2):
    res = {}
    for key, value in dict1.items():
        if key in dict2:
            res[key] = merge_nested_dicts(dict1[key], dict2[key])
            del dict2[key]
        else:
            res[key]=value
    res.update(dict2)
    return res