常用于矩阵求和计算,以下用法分为三种情况来介绍!
格式:np.sum(a)
np.sum(a, axis=0) ------->列求和
np.sum(a, axis=1) ------->行求和
注意:特别注意后两种用法。
In : a = np.array([[0, 2, 1], [3, 5, 6], [0, 1, 1]]) #得到一个矩阵a
Out: [[0 2 1]
[3 5 6]
[0 1 1]]
In : a.sum() #将a中所有元素求和 0+2+1+3+5+6+0+1+1=19
Out: 19
In : np.sum(a,axis=1) # 求出每行元素的和
Out: [ 3 14 2]
In : np.sum(a,axis=0) # 求出每列元素的和
Out: [3 8 8]