列表清单的总和

Question

我正在寻找python中的方法来汇总只包含整数的列表.我看到该方法sum()仅适用于列表,但不适用于列表列表.有什么适合我吗？

谢谢

Answer 1

您可以sum()在此处使用生成器表达式:

In [18]: lis = [[1, 2], [3, 4], [5, 6]]

In [19]: sum(sum(x) for x in lis)
Out[19]: 21

要么:

In [21]: sum(sum(lis, []))
Out[21]: 21

timeit 比较:

In [49]: %timeit sum(sum(x) for x in lis)
100000 loops, best of 3: 2.56 us per loop

In [50]: %timeit sum(map(sum, lis))
100000 loops, best of 3: 2.39 us per loop

In [51]: %timeit sum(sum(lis, []))
1000000 loops, best of 3: 2.21 us per loop

In [52]: %timeit sum(chain.from_iterable(lis))       # winner 
100000 loops, best of 3: 1.43 us per loop

In [53]: %timeit sum(chain(*lis))
100000 loops, best of 3: 1.55 us per loop

Answer 2

import itertools

sum(itertools.chain.from_iterable([[1,2],[3,4],[5,6]]))

itertools.chain扁平化一层可迭代（这是您在这里所需要的全部），因此sum获取带有子列表的列表。