Python NumPy sum方法用法及代碼示例

例子

基本用法

np.sum([1,2,3,4])



10

計算二維數組的總和

考慮以下二維數組：

a = np.array([[1,2],[3,4]])
a



array([[1, 2],
       [3, 4]])

所有值

np.sum(a)



10

按列

np.sum(a, axis=0)



array([4, 6])

逐行

np.sum(a, axis=1)



array([3, 7])

指定輸出數組

a = np.zeros(2)
np.sum([[1,2],[3,4]], axis=1, out=a)   # row-wise summation
a



array([3., 7.])

在這裏，我們將結果輸出到數組 a 。

指定初始值

np.sum([1,2,3], initial=10)



16

在這裏，由於我們設置了初始值 10，所以我們有 10+1+2+3=16 。

指定布爾掩碼

np.sum([4,5,6,7], where=[False, True, True, False])



11

這裏，在總和的計算中僅包括第二個和第三個值。