如何对二维列表的列求和？

Question

假设我有一个这样的Python二维列表：

my_list =  [ [1,2,3,4],
             [2,4,5,6] ]

我可以通过列表推导式来计算每一行的总和：

row_totals = [ sum(x) for x in my_list ]

那么我能不能不使用双重for循环来计算每一列的总和呢？也就是说，我想得到这个列表：

[3,6,8,10]

Answer 1

解决方案 map(sum,zip(*my_list)) 是最快的。

不过，如果你需要保留这个列表，使用 [x + y for x, y in zip(*my_list)] 会更快。

这个测试是在 Python 3.1.2 64 位版本中进行的。

>>> import timeit
>>> my_list = [[1, 2, 3, 4], [2, 4, 5, 6]]
>>> t1 = lambda: [sum(x) for x in zip(*my_list)]
>>> timeit.timeit(t1)
2.5090877081503606
>>> t2 = lambda: map(sum,zip(*my_list))
>>> timeit.timeit(t2)
0.9024796603792709
>>> t3 = lambda: list(map(sum,zip(*my_list)))
>>> timeit.timeit(t3)
3.4918002495520284
>>> t4 = lambda: [x + y for x, y in zip(*my_list)]
>>> timeit.timeit(t4)
1.7795929868792655

Answer 2

>>> map(sum,zip(*my_list))
[3, 6, 8, 10]

>>> from itertools import imap, izip
>>> imap(sum,izip(*my_list))
<itertools.imap object at 0x00D20370>
>>> list(_)
[3, 6, 8, 10]

或者用itertools的对应方法

Answer 3

使用 zip 函数

col_totals = [ sum(x) for x in zip(*my_list) ]