字典中的字典中的值求和

3

我得到了这样的东西:

> d1 = {'System tests': {'failed': 5, 'passed': 0, 'total': 5},
       'Func tests': {'failed': 5, 'passed': 0, 'total': 5}}

> d2 = {'System tests': {'failed': 1, 'passed': 1, 'total': 2}, 
        'Func tests': {'failed': 3, 'passed': 2, 'total': 5}}

> d3 = {'System tests': {'failed': 0, 'passed': 0, 'total': 0}, 
        'Func tests': {'failed': 1, 'passed': 0, 'total': 1}}

我希望将'failed', 'passed'和'total'的值汇总到一个字典中。输出应该像这样:
```python { 'failed': 2, 'passed': 8, 'total': 10 } ```
d4 = {'System tests': {'failed': 6, 'passed': 1, 'total': 7}, 
       'Func tests': {'failed': 9, 'passed': 2, 'total': 11}

什么是最简单的解决方案来完成这件事?

我可以使用基本库,但不包括collections。

解决方案必须是通用的,例如如果将来出现其他字典。


2
欢迎来到StackOverflow!请参观我们的tour,详细了解如何提问,并提供一个最小、完整和可验证的示例,以重现您的问题。 - hiro protagonist
5个回答

5

输入:

d1 = {'a': 100, 'b': 200, 'c':300}
d2 = {'a': 300, 'b': 200, 'd':400}
d = {k : d1.get(k, 0) + d2.get(k,0) for k in set(d1.keys()) | set(d2.keys())}

输出:

{'a': 400, 'b': 400, 'c': 300, 'd': 400}

4

使用 Counterdefaultdict 是相当简单的,如下所示:

代码:

d4 = defaultdict(Counter)
for d in d1, d2, d3:
    for k, subd in d.items():
        d4[k].update(subd)

测试代码:

d1 = {'System tests': {'failed': 5, 'passed': 0, 'total': 5},
        'Func tests': {'failed': 5, 'passed': 0, 'total': 5}}

d2 = {'System tests': {'failed': 1, 'passed': 1, 'total': 2},
        'Func tests': {'failed': 3, 'passed': 2, 'total': 5}}

d3 = {'System tests': {'failed': 0, 'passed': 0, 'total': 0},
        'Func tests': {'failed': 1, 'passed': 0, 'total': 1}}

from collections import Counter, defaultdict

d4 = defaultdict(Counter)
for d in d1, d2, d3:
    for k, subd in d.items():
        d4[k].update(subd)
print(d4)

结果:

defaultdict(<class 'collections.Counter'>, {
    'System tests': Counter({'total': 7, 'failed': 6, 'passed': 1}), 
    'Func tests': Counter({'total': 11, 'failed': 9, 'passed': 2})
})

2
这将为您提供所需的输出,无需使用任何库。
d4 = {}

for d in d1, d2, d3:
    for test, results in d.items():
        if test not in d4:
            d4[test] = {}
        for key, value in results.items():
            if key in d4[test]:
                d4[test][key] += value
            else:
                d4[test][key] = value

结果:

{'System tests': {'failed': 6, 'passed': 1, 'total': 7}, 'Func tests': {'failed': 9, 'passed': 2, 'total': 11}}

这正是我所需要的! - kk_pl

0

如果您无法使用 collections.Counter,另一种解决方案是使用 reduce

from functools import reduce # reduce was moved here in Python3

def add_dicts(dict_list):

    def add(fst, snd):
        return {k: fst[k] + snd[k] for k in fst}

    return reduce(add, dict_list[1:], dict_list[0])

在你的代码中使用它会像这样。
dict_list = [d1, d2, d3]

d4 = {}

for k in dict_list[0]:
    d4[k] = add_dicts([d[k] for d in dict_list])

尽管如此,这假定你的所有dict都格式正确。

0
你可以使用 zipdict.items 来实现:
d1 = {'System tests': {'failed': 5, 'passed': 0, 'total': 5},
   'Func tests': {'failed': 5, 'passed': 0, 'total': 5}}
d2 = {'System tests': {'failed': 1, 'passed': 1, 'total': 2}, 
    'Func tests': {'failed': 3, 'passed': 2, 'total': 5}}
d3 = {'System tests': {'failed': 0, 'passed': 0, 'total': 0}, 
    'Func tests': {'failed': 1, 'passed': 0, 'total': 1}}

def combine_dicts(*d):
   return {a:{i:sum(h[i] for h in [b, c, d]) for i in ['failed', 'passed', 'total']} for [a, b], [_, c], [_, d] in zip(*d)}

print(combine_dicts(d1.items(), d2.items(), d3.items()))

输出:

{'System tests': {'failed': 6, 'total': 7, 'passed': 1}, 'Func tests': {'failed': 9, 'total': 11, 'passed': 2}}

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接