为了账号安全,请及时绑定邮箱和手机立即绑定

如何将嵌套字典转换为平面字典并使用重复键对值求和?

如何将嵌套字典转换为平面字典并使用重复键对值求和?

温温酱 2022-05-24 13:05:09
这是我的字典: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}(显然没有意义,但为了说明这一点)寻找最干净的方法来做到这一点
查看完整描述

2 回答

?
冉冉说

TA贡献1877条经验 获得超1个赞

另一种使用方式collections.Counter:


from collections import Counter


sum(map(Counter, d.values()), Counter())

输出:


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


查看完整回答
反对 回复 2022-05-24
?
函数式编程

TA贡献1807条经验 获得超9个赞

您可以使用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}


查看完整回答
反对 回复 2022-05-24
  • 2 回答
  • 0 关注
  • 106 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
微信客服

购课补贴
联系客服咨询优惠详情

帮助反馈 APP下载

慕课网APP
您的移动学习伙伴

公众号

扫描二维码
关注慕课网微信公众号