在python中查找多维数组中列总和的最佳方法？ [复制]

Question

我以元组数组的形式从 SQL 查询中获取输出。数组看起来像这样-

(1, 100)
(2, 150)
(3, 200)

我想分别求各列的总和。因此，我想为第 1 列获取值 1+2+3= 6，并为第 2 列获取值 100+150+200= 450。

当我使用内置 var1=sum(arr[0]) 时，我得到 101，这是第一个 row 而不是 column 的总和。

如何以最简单的方式将第一列相加？

Answer 1

您可以使用zip 和list comprehension：

>>> lst = [
...     (1, 100),
...     (2, 150),
...     (3, 200)
... ]
>>> list(zip(*lst))  # zip(*lst) groups the items that you want to sum
[(1, 2, 3), (100, 150, 200)]
>>>
>>> [sum(x) for x in zip(*lst)]
[6, 450]
>>>