嵌套列表中第一个值的总和

Question

在传统的python中,sum函数给出了一个总和list:

sum([0,1,2,3,4])=10

另一方面,如果你有一个嵌套列表如下:

sum([[1,2,3],[4,5,6],[7,8,9]])

我们发现错误:

Traceback (most recent call last):
  File "<input>", line 1, in <module>
TypeError: unsupported operand type(s) for +: 'int' and 'list'

除此之外,我们怎样才能sum在嵌套列表中找到第一个值(索引0)？如:

something([[1,2,3],[4,5,6],[7,8,9]])=12

Answer 1

要获得生成器表达式所需的所有第一个元素的总和

>>> a = [[1,2,3],[4,5,6],[7,8,9]]
>>> sum(i[0] for i in a)
12

你得到的unsupported operand type(s) for +: 'int' and 'list'是因为你试图添加三个不是所需行为的列表.

如果你想要一个第一个元素的列表然后找到它们的总和,你可以尝试一个列表理解

>>> l = [i[0] for i in a]
>>> l
[1, 4, 7]
>>> sum(l)
12

或者您可以调用__next__方法,因为列表是可迭代的(如果Py3)

>>> sum(zip(*a).__next__())
12