**
np.sum(array, axis)
**
- 指定数组元素求和, 如果axis为None,则数组所有元素求和,得到一个值。如果指定axis,则按指定轴求和。
- 参数:
array : 数组形状
axis: 默认None, 或者int型
keepdims :是否保持计算后数据维度 - 例子
(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])
>>>