我有一本计数器字典,例如:
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 中执行此操作的最简单/惯用的方法是什么?
我有一本计数器字典,例如:
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 中执行此操作的最简单/惯用的方法是什么?
使用嵌套推导:
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()))
我更喜欢第一种方式。
sum(sum(c.values()) for c in numbers.values())
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()))