3

我有一本计数器字典,例如:

from collections import Counter, defaultdict

numbers = defaultdict(Counter)

numbers['a']['first'] = 1
numbers['a']['second'] = 2
numbers['b']['first'] = 3

我想得到总和:1+2+3 = 6

在 python 3 中执行此操作的最简单/惯用的方法是什么?

4

3 回答 3

7

使用嵌套推导:

sum(x for counter in numbers.values() for x in counter.values())

或者先对计数器求和(从一个空的开始),然后是它们的值:

sum(sum(numbers.values(), Counter()).values())

或者首先是每个计数器的值,然后是中间结果:

sum(sum(c.values()) for c in numbers.values())

或使用链:

from itertools import chain
sum(chain.from_iterable(d.values() for d in numbers.values()))

我更喜欢第一种方式。

于 2013-03-11T23:47:33.880 回答
2
sum(sum(c.values()) for c in numbers.values())
于 2013-03-11T23:49:18.443 回答
1
from itertools import chain
sum(chain.from_iterable(d.values() for d in numbers.values()))
# outputs: 6

在 python 2.x 中的性能使用方面.itervalues(),避免了构建中间列表(适用于此处的所有解决方案)。

sum(chain.from_iterable(d.itervalues() for d in numbers.itervalues()))
于 2013-03-11T23:47:04.260 回答