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

时间:2015-02-01 16:08:48

标签: python list sum list-comprehension nested-lists

在传统的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

6 个答案:

答案 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