python - 如何将嵌套字典变成平面字典并用重复键对值求和？

Question

这是我的字典:

d = {'dicta':{'a':1,'b':2,'c':3}, 'dictb':{'a':2,'b':3,'c':1}, 'dictc':{'a':2,'b':5,'c':9,'d':10}}

{'dicta': {'a': 1, 'b': 2, 'c': 3}, 'dictb': {'a': 2, 'b': 3, 'c': 1}, 'dictc': {'a': 2, 'b': 5, 'c': 9, 'd': 10}}

我想做的是将这些嵌套的字典组合成一个平面字典，并对值求和得到:

{'a': 5, 'b': 10, 'c': 13, 'd': 10}

我已经可以使用下面的代码实现这一点:

out = {}
for k,v in [[k2,d[k1][k2]] for k1 in d for k2 in d[k1]]:
    if k in out.keys():
        out[k] = out[k] + v
    else:
        out[k] = v

但是如您所见，它并不是很优雅。一定会有更好的办法! (雷蒙德·海廷格 :D)。

我也可以做一个字典理解，显然每个键只保留一个条目:

out = {k2:d[k1][k2] for k1 in d for k2 in d[k1]}

输出:{'a': 2, 'b': 5, 'c': 9, 'd': 10}

或者一个列表理解，它至少给了我想要组合的一切:

out = [{k2:d[k1][k2]} for k1 in d for k2 in d[k1]]

输出:[{'a': 1}, {'b': 2}, {'c': 3}, {'a': 2}, {'b': 3}, { 'c': 1}, {'a': 2}, {'b': 5}, {'c': 9}]

这感觉就像我越来越接近......

有没有办法以一种我只是不知道的非常优雅的方式按照我想要的方式组合这些？

类似于:out = {k2:d[k1][k2] for k1 in d for k2 in d[k1] with combined = True}(显然没有意义，但说明这一点)

寻找最干净的方法来做到这一点

Answer 1

你可以使用collections.defaultdict:

from collections import defaultdict
d = {'dicta':{'a':1,'b':2,'c':3}, 'dictb':{'a':2,'b':3,'c':1}, 'dictc':{'a':2,'b':5,'c':9,'d':10}}
_d = defaultdict(int)
for a in d.values():
  for c, j in a.items():
     _d[c] += j

print(dict(_d))

输出:

{'a': 5, 'b': 10, 'c': 13, 'd': 10}

使用 itertools.groupby 的更短解决方案:

from itertools import groupby
r = sorted([i for b in d.values() for i in b.items()], key=lambda x:x[0])
result = {a:sum(c for _, c in b) for a, b in groupby(r, key=lambda x:x[0])}

输出:

{'a': 5, 'b': 10, 'c': 13, 'd': 10}