嵌套列表中的第一个值的总和
问题内容:
在传统的python中,该sum
函数给出a的总和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
问题答案:
要获得所有第一个元素的总和,您需要有一个生成器表达式
>>> 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