Python：将两个嵌套字典合并，其中顶级键的值为字典

Question

我想要合并两个嵌套字典

d1 = {"admin": {"key1": "v2"}}
d2 = {"admin": {"key2": "v3"},
      "user": {"something": "else"}}

这应该合并为：

d = {"admin": {"key1": "v2",
               "key2": "v3"},
     "user": {"something": "else"}}

除了迭代第一个键之外，还有其他简单的方法吗？

- cornelinux

Answer 1

您可以使用defaultdict将第一层的每个值作为dict。然后，您可以使用字典中的update()方法。

>>> from collections import defaultdict
>>> d = defaultdict(dict)
>>> d1 = {"admin": {"key1": "v2"}}
>>> d2 = {"admin": {"key2": "v3"},
...       "user": {"something": "else"}}
>>> for dd in [d1,d2]:
...     for k,v in dd.items():
...         d[k].update(v)
... 
>>> d
defaultdict(<type 'dict'>, {'admin': {'key2': 'v3', 'key1': 'v2'}, 'user': {'something': 'else'}})

或者更紧凑的形式，您可以使用map()...真的我不喜欢它，但是这是一个机会

>>> for dd in [d1,d2]:
...     map(lambda x:d[x[0]].update(x[1]),dd.items())

- Michele d'Amico

Answer 2

如果只有一层嵌套：

>>> d1 = {"admin": {"key1": "v2"}}
>>> d2 = {"admin": {"key2": "v3"},
...       "user": {"something": "else"}}
>>> keys = list(d1) + list(d2)
>>> d = {k: dict(d1.get(k, {}).items() + d2.get(k, {}).items()) for k in keys}
>>> d
{'admin': {'key1': 'v2', 'key2': 'v3'}, 'user': {'something': 'else'}}

较深层次的嵌套将需要递归。

- wim

---

太酷了。我不知道我可以对字典进行“列表”推导。 - cornelinux

Answer 3

from itertools import chain
from collections import defaultdict
new_d = defaultdict(dict)

for k,v in chain(d1.iteritems(),d2.iteritems()):
    new_d[k].update(v)
print(new_d)
defaultdict(<type 'dict'>, {'admin': {'key2': 'v3', 'key1': 'v2'}, 'user': {'something': 'else'}})

- Padraic Cunningham

Answer 4

from copy import deepcopy

d1 = {"admin": {"key1": "v2"}}
d2 = {"admin": {"key2": "v3"},
      "user": {"something": "else"}}

def merge(one, two):
    if isinstance(two, dict):
        result = deepcopy(one)
        for key, value in two.iteritems():
            if key in result and isinstance(result[key], dict):
                result[key] = merge(result[key], value)
            else:
                result[key] = deepcopy(value)
        return result
    return two

print merge(d1, d2)

输出

{'admin': {'key2': 'v3', 'key1': 'v2'}, 'user': {'something': 'else'}}

- Ngenator