在传统的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
答案 0 :(得分:7)
获取生成器表达式所需的所有第一个元素的总和
>>> 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
答案 1 :(得分:1)
或者您可以使用zip
:
>>> l=[[1,2,3],[4,5,6],[7,8,9]]
>>> sum(zip(*l)[0])
12
答案 2 :(得分:0)
您可以创建一个函数来查找嵌套列表的总和:
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也有一些简洁的答案,检查出来!
答案 3 :(得分:0)
对于python初学者:
通过正常for
循环并使用try和except来处理异常,当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
>>>
答案 4 :(得分:0)
>>> sum(map(lambda x:x[0],[[1,2,3],[4,5,6],[7,8,9]]))
12
答案 5 :(得分:0)
使用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