[英]How do I sum the columns in 2D list?
說我有一個Python 2D列表如下:
my_list = [ [1,2,3,4],
[2,4,5,6] ]
我可以通過列表理解得到行總數:
row_totals = [ sum(x) for x in my_list ]
我可以列總計沒有雙for
循環? 即,獲取此列表:
[3,6,8,10]
使用拉鏈
col_totals = [ sum(x) for x in zip(*my_list) ]
>>> map(sum,zip(*my_list))
[3, 6, 8, 10]
或者等效的itertools
>>> from itertools import imap, izip
>>> imap(sum,izip(*my_list))
<itertools.imap object at 0x00D20370>
>>> list(_)
[3, 6, 8, 10]
解決方案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
[x + y for x, y in zip(*my_list)]
import numpy as np
my_list = np.array([[1,2,3,4],[2,4,5,6]])
[ sum(x) for x in my_list.transpose() ]
出[*]:[3,6,8,10]
my_list.sum(axis = 0)
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.