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

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)

Answer 2

对于任何矩阵(或其他雄心勃勃的数字)操作,我建议调查NumPy.

用于求解问题中所示轴的数组之和的样本将是:

>>> from numpy import array
>>> data = array([[3,7,2],
...     [1,4,5],
...     [9,8,7]])
>>> from numpy import sum
>>> sum(data, 0)
array([13, 19, 14])

这是numpy的sum函数文档:http://docs.scipy.org/doc/numpy/reference/generated/numpy.sum.html#numpy.sum

尤其是第二个参数很有意思,因为它允许轻松指定应该总结的内容:所有元素或仅可能是n维数组的特定轴(如).

Answer 3

这将为您提供每个子列表的总和

data = [[3,7,2],[1,4],[9,8,7,10]]
list(map(sum, data))
[12, 5, 34]

如果你想总结所有元素并得到一个总和,那么使用它

data = [[3,7,2],[1,4],[9,8,7,10]]
sum(sum(data, []))
51