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

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'

除此之外，我们如何找到嵌套列表中第一个值（索引为0）的总和？例如：

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

- user2592835

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

如果您正在使用 Python 3，则可以将 __next__ 方法作为列表的可迭代对象进行调用。

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

- Bhargav Rao

Answer 2

或者您可以使用zip：

>>> l=[[1,2,3],[4,5,6],[7,8,9]]
>>> sum(zip(*l)[0])
12

- Mazdak

Answer 3

你可以创建一个函数来查找嵌套列表的总和：

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也有一些不错的答案，去看看吧！

- m0bi5

Answer 4

针对Python初学者：

通过普通的for循环，并使用try和except来处理当列表为空时的异常情况。

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

- Vivek Sable

---

不要使用旧式的 except IndexError, e，它只是在 Python 2 中为向后兼容而存在（在 Python 3 中已被删除），请使用：except IndexError as e。顺便说一句，如果您对异常什么都不做，可以简单地使用：except IndexError。 - Ashwini Chaudhary

@AshwiniChaudhary：是的，谢谢您审查我的代码，现在已更新。 - Vivek Sable

Answer 5

>>> sum(map(lambda x:x[0],[[1,2,3],[4,5,6],[7,8,9]]))
12

- Hackaholic

Answer 6

使用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

- Akavall