列表列表的总和；返回总和清单

Question

让 data = [[3,7,2],[1,4,5],[9,8,7]]

假设我要对列表中每个列表的索引元素求和，例如在矩阵列中添加数字以获得单个列表。我假设数据中的所有列表的长度均相等。

    print foo(data)

   [[3,7,2],
    [1,4,5],
    [9,8,7]]
    _______
 >>>[13,19,14]

如何遍历列表列表而不会出现索引超出范围错误？也许lambda？谢谢！

Answer 1

您可以尝试以下方法：

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)

在这种不相等长度的列表的情况下，可以使用itertools.izip_longest具有fillvalue的0-这主要填充缺失与指数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)