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

这里，在总和的计算中仅包括第二个和第三个值。