Python 在numpy数组中计算和

我想知道如何在带有循环的3x3

NumPy

数组中获取值的总和。 例如：

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

---

输出应该是

[6,15,24]

尝试使用

np.array.sum

和

轴

参数作为一个参数：

arr.sum(1)

例如：

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

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

或者，如果是列表：

np.sum(array,1)

例如：

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

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

检查这个