[英]Sum nested key values of dict
这是我在Python 2.7中的示例字典:
sample = {'T1': {'P1': 20, 'P2': 100}, 'T2': {'P1': 60, 'P2': 50}}
我试图用键'P1'和'P2'来总结所有值,得到这样的结果:
reqResult = [80,150]
我该怎么做?
非常感谢。
您可以使用
>>> d = {'T1': {'P1': 20, 'P2': 100}, 'T2': {'P1': 60, 'P2': 50}}
>>> map(sum, zip(*[x.values() for x in d.values()]))
[150, 80]
这将首先计算内在的dicts,而不是取出它们的值并将它们压缩,最后将它们全部加起来。
或者,定义自定义函数并使用它:
>>> d = {'T1': {'P1': 20, 'P2': 100}, 'T2': {'P1': 60, 'P2': 50}}
>>> def sigma(list_of_dicts):
... result = []
... keys = list_of_dicts[0].keys()
... for key in keys:
... result.append(sum(x[key] for x in list_of_dicts))
... return result
...
>>> print sigma(d.values())
[150, 80]
从您问题的标签,您似乎正在寻找列表理解来做到这一点。 通常情况下,它们可能有点难以阅读 - 但这里有一个:
from collections import Counter
sample = {'T1': {'P1': 20, 'P2': 100}, 'T2': {'P1': 60, 'P2': 50}}
reqResult = [v[1] for v in sorted(reduce(lambda c, d: (c.update(d), c)[1],
sample.values(), Counter()).items())]
print reqResult # --> [80, 150]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.