在传统的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'
除此之外,我们如何在嵌套列表中找到第一个值(索引0)的sum
?如:
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
或者你可以使用zip
:
>>> l=[[1,2,3],[4,5,6],[7,8,9]]
>>> sum(zip(*l)[0])
12
您可以创建一个函数来查找嵌套列表的总和:
def nested_sum(par):
total = 0
for k in par:
if isinstance(k, list): # checks if `k` is a list
total += nested_sum(k)
else:
total += k
return total
@Kasara和@Bhargav也有一些简洁的答案,检查出来!
>>> sum(map(lambda x:x[0],[[1,2,3],[4,5,6],[7,8,9]]))
12
使用numpy
这样的东西很容易:
In [16]: import numpy as np
In [17]: a = [[1,2,3],[4,5,6],[7,8,9]]
In [18]: my_array = np.array(a)
In [19]: my_array[:,0].sum()
Out[19]: 12
对于python初学者:
通过正常的for
循环并使用try和除了在bested列表为空时处理异常。
>>> l = [[1,2,3],[4,5,6],[7,8,9], []]
>>> result = 0
>>> for i in l:
... try:result += i[0]
... except IndexError:pass
...
>>> result
12
>>>