axis = 0,对横轴(第0维)进行操作,运算的方向为纵向运算。即求列和、最大、最小等
axis = 1,对纵轴(第1维)进程操作,运算的方向为横向运算。即求行和、最大、最小等
import numpy as np
import numpy as np
data = np.array([[1,1,1,1],
[2,2,2,2],
[3,3,3,3],
[4,4,4,4]])
Out[1]:
array([[1, 1, 1, 1],
[2, 2, 2, 2],
[3, 3, 3, 3],
[4, 4, 4, 4]])
#对横轴进行操作,即纵向运算
np.sum(data,axis=0)
Out[2]:
array([10, 10, 10, 10])
#对纵轴进行操作,即横向运算
np.sum(data,axis=1)
Out[3]:
array([ 4, 8, 12, 16])
参考文献:
https://blog.csdn.net/brucewong0516/article/details/79030994