import numpy as np
a = np.array([[1,2],[3,4],[5,6]])
print(a.shape)
得到输出 (3, 2) , 可知python序列从外到内,维度依次增加
a_mean_0 = np.mean(a, axis=0)
print(a_mean_0)
得到输出 [3. 4.]
a_mean_1 = np.mean(a, axis=1)
print(a_mean_1)
得到输出 [1.5 3.5 5.5]
由此可见
axis等于几,就是在第几维度进行操作,也就是将第几维度压缩至一维。