np.sum()

**

np.sum(array, axis)

**

  1. 指定数组元素求和, 如果axis为None,则数组所有元素求和,得到一个值。如果指定axis,则按指定轴求和。
  2. 参数:
    array : 数组形状
    axis: 默认None, 或者int型
    keepdims :是否保持计算后数据维度
  3. 例子
    (1) axis默认值
>>> a = np.arange(6).reshape(2,3)
>>> a
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.sum(a)
15
>>> 

(2) 指定轴求和,
np.sum(a, axis=0, keepdims=True) , 按列求和

>>> np.sum(a, axis=0, keepdims=True)
array([3, 5, 7])
>>> 

np.sum(a, aixs=1, keepdims=Ture), 按行求和

>>> np.sum(a, axis=1, keepdims=True)
array([3],
      [12])
>>> 

你可能感兴趣的:(python学习)