计算numpy数组中的总和

Question

我想知道如何通过循环在3x3 NumPy数组中获取值的总和。 例如：

[[1,2,3],[4,5,6],[7,8,9]]

输出应为[6,15,24]。

Answer 1

尝试将np.array.sum参数和axis一起使用：

arr.sum(1)

示例：

>>> arr=np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> arr.sum(1)
array([ 6, 15, 24])

或者如果它是一个列表：

np.sum(array,1)

示例：

>>> arr=[[1,2,3],[4,5,6],[7,8,9]]
>>> np.sum(arr,1)
array([ 6, 15, 24])
>>>