- 降维求和
A = np.arange(20).reshape(5, 4)
A=array([[ 0., 1., 2., 3.],
[ 4., 5., 6., 7.],
[ 8., 9., 10., 11.],
[12., 13., 14., 15.],
[16., 17., 18., 19.]])
A.shape, A.sum()
((5, 4), array(190.))
设置axis=0,说明要把每行给压缩,即求和所有行的元素A_sum_axis0 = A.sum(axis=0)
A_sum_axis0, A_sum_axis0.shape
理解记忆:axis = 0按照行,可以理解为把“行”给抹去只剩1行,也就是上下压扁。
axis = 1按照列,可以理解为把“列”给抹去只剩1列,也就是左右压扁
2.非降维求和:在求和时保持轴数不变会有用