通用函数，用于合并嵌套字典中不同键数的不同数量的字典

Question

我想创建一个通用函数来合并python中嵌套字典中具有不同键的不同数量的嵌套字典。但是，如果在嵌套的字典中有相同的键，下面的函数可以正常工作:

data = [dict1, dict2, dict3]
result = {k: {ik: [] for ik in dict1[k].keys()} for k in dict1.keys()}
for entry in data:
  for key, value in entry.items():
    for inner_key, inner_value in value.items():
        result[key][inner_key].append(inner_value)

我的输入字典是:

dict1 = {"canada":{'america':189,'asia':175,'europe': 186},
     "norway":{'australia': 123,'africa':124,'brazil':125}}
dict2=  {"canada":{'florida':13,'asia':14,'europe': 15},
     "norway":{'california': 17,'africa':18,'brazil':19}}
dict3=  {"canada":{'africa':127,'asia':256,'europe': 16, 'brazil':15},
     "norway":{'australia': 17,'africa':18,'brazil':19}}
# Expected Output
{'canada': {'africa': 127,
'america': 189,
'asia': [175, 14, 256],
'brazil': 15,
'europe': [186, 15, 16],
'florida': 13},
'norway': {'africa': [124, 18, 18],
'australia': 123,
'brazil': [125, 19, 19],
'california': 17}}

Answer 1

一种选择是使用dict.setdefault来更新内部字典。

out = {}
for k in dict1.keys() | dict2.keys() | dict3.keys():
    inner = {}
    for d in [dict1, dict2, dict3]:
        for i,j in d.get(k, {}).items():
            inner.setdefault(i, []).append(j)
    out[k] = inner

如果字典中的其他键相同，我们可以稍微简化一下。

同样，如果你坚持使用整型(而不是列表)的单例列表，你可以尝试以下操作(但我不认为混合数据类型是一个好主意，因为之后你需要检查值是整型还是列表):

def merge_dicts(dicts):
    out = {}
    for k in dicts[0]:
        inner = {}
        for d in dicts:
            for i,j in d.get(k, {}).items():
                if i in inner and isinstance(inner[i], list):
                    inner[i].append(j)
                elif i in inner:
                    inner[i] = [inner[i], j]
                else:
                    inner[i] = j
        out[k] = inner
    return out

输出:

>>> out = merge_dicts([dict1,dict2,dict3])
{'canada': {'america': 189,
  'asia': [175, 14, 256],
  'europe': [186, 15, 16],
  'florida': 13,
  'africa': 127,
  'brazil': 15},
 'norway': {'australia': [123, 17],
  'africa': [124, 18, 18],
  'brazil': [125, 19, 19],
  'california': 17}}