你可以试试这个:
In [9]: l = [[3,7,2],[1,4,5],[9,8,7]]
In [10]: [sum(i) for i in zip(*l)]
Out[10]: [13, 19, 14]
这使用zip和*的组合来打包列表,然后根据索引压缩项目。然后,您使用列表解析来迭代类似索引的组,对它们进行求和并返回其“原始”位置。
希望可以让这一点更加清晰,当您重复使用zip(* l)时,会发生什么:
In [13]: for i in zip(*l):
....: print i
....:
....:
(3, 1, 9)
(7, 4, 8)
(2, 5, 7)
在长度不等的列表的情况下,您可以使用filltools为0的itertools.izip_longest – 这基本上将缺少的索引填充为0,允许您将所有列列入:
In [1]: import itertools
In [2]: l = [[3,7,2],[1,4],[9,8,7,10]]
In [3]: [sum(i) for i in itertools.izip_longest(*l, fillvalue=0)]
Out[3]: [13, 19, 9, 10]
在这种情况下,这是在izip_longest上如何迭代:
In [4]: for i in itertools.izip_longest(*l, fillvalue=0):
...: print i
...:
(3, 1, 9)
(7, 4, 8)
(2, 0, 7)
(0, 0, 10)