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

Question

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

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

我想分别找到列的总和。因此，我希望第1列的值为1 + 2 + 3 = 6，第2列的值为100 + 150 + 200 = 450。

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

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

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