Numpy轴的概念

numpy中轴的概念是可以认为是数组的维度的概念

a = np.arange(24).reshape(2,3,4)
a

array([[[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]],

[[12, 13, 14, 15],
[16, 17, 18, 19],
[20, 21, 22, 23]]])

np.max(a,axis=0)

array([[12, 13, 14, 15],
[16, 17, 18, 19],
[20, 21, 22, 23]])

np.max(a,axis=1)

array([[ 8, 9, 10, 11],
[20, 21, 22, 23]])

np.max(a,axis=2)

array([[ 3, 7, 11],
[15, 19, 23]])