有没有Python方法可以用另一个字典扩展一个字典(保留所有值)

Question

给定字典：

d1={'a':'a','b':'b','c':'c'}
d2={'b':'a','c':['d','f','g'],'e':'e'}

这两个字典可以合并所有公共键并保留所有值吗？即给出输出：

> print(d1.extend(d2))
{'a':'a','b':['b','a'],'c':['c','d','f','g'],'e':'e'}

我想出了以下方法，这似乎很有效，但非常不符合Python。

def extend(d1, d2):
return_dict={}
for key, value in d1.items():
if key in d2:
value_d2=d2[key]
if value == value_d2:
continue
if type(value) == list and type(value_d2) == list:
value.extend(value_d2)
return_dict[key]=value
elif type(value) == list and type(value_d2) != list:
tmp=[value_d2]
tmp.extend(value)
return_dict[key]=tmp
elif type(value) != list and type(value_d2) == list:
tmp=[value]
tmp.extend(value_d2)
return_dict[key]=tmp
elif type(value) != list and type(value_d2) != list:
return_dict[key]=[value] + [value_d2]
else:
return_dict[key]=value
for key, value in d2.items():
if key not in return_dict:
return_dict[key]=value
return return_dict

(最后一个elif应该是else，但我认为这样更可读(

编辑：

是否可以保留所有键，但删除重复的值，而不是保留所有值？即

d1={'a':'a','b':'b','c':'c'}
d2={'b':'b','c':['d','f','g'],'e':'e'}
> print(d1.extend(d2))
{'a':'a','b':'b','c':['c','d','f','g'],'e':'e'}

Answer 1

使用collections.defaultdict作为临时存储，如下所示：

from collections import defaultdict
d1 = {'a': 'a', 'b': 'b', 'c': 'c'}
d2 = {'b': 'a', 'c': ['d', 'f', 'g'], 'e': 'e'}
tmp = defaultdict(list)
for d in [d1, d2]:
for k, v in d.items():
if isinstance(v, list):
tmp[k].extend(v)
else:
tmp[k].append(v)

res = { k : v if len(v) > 1 else v[0] for k, v in tmp.items()}
print(res)

输出

{'a': 'a', 'b': ['b', 'a'], 'c': ['c', 'd', 'f', 'g'], 'e': 'e'}

另一种选择，也使用defaultdict，是这样做：

tmp1 = defaultdict(list)
tmp2 = defaultdict(list)
tmp1.update(d1)
tmp2.update(d2)
tmp = {key: [*tmp1[key], *tmp2[key]] for key in tmp1.keys() | tmp2.keys()}
res = {k: v if len(v) > 1 else v[0] for k, v in tmp.items()}
print(res)

这两种方法都适用于Python 3.7。

更新

正如@ShadowRanger所提到的，你可以使用一个集合，而不是列表：

tmp1 = defaultdict(set)
tmp2 = defaultdict(set)
tmp1.update(d1)
tmp2.update(d2)
tmp = {key: [*tmp1[key], *tmp2[key]] for key in tmp1.keys() | tmp2.keys()}
res = {k: v if len(v) > 1 else v[0] for k, v in tmp.items()}
print(res)

Answer 2

您可以将助手函数safe_combine与Python 3.8+中提供的dict.union运算符|结合使用：

from __future__ import annotations

d1 = {'a': 'a', 'b': 'b', 'c': 'c'}
d2 = {'b': 'a', 'c': ['d', 'f', 'g'], 'e': 'e'}

def safe_combine(o1: str | list, o2: str | list):
return (o1 if isinstance(o1, list) else [o1]) 
+ (o2 if isinstance(o2, list) else [o2])

merged = {k: safe_combine(d1[k], d2[k]) if k in d1 and k in d2 else v
for k, v in (d1 | d2).items()}
print(merged)

输出：

{'a': 'a', 'b': ['b', 'a'], 'c': ['c', 'd', 'f', 'g'], 'e': 'e'}

---

NB：对于3.8之前的Python版本，可以使用{**d1, **d2}语法而不是(d1 | d2)。