numpy2.1: axis=0、1

二维数组

axis=0 : 跨行操作，按列计算，返回一行所拥有的元素数
axis=1 : 跨列操作，按行计算，返回一列所拥有的元素数

import numpy as np
arr = np.linspace(1,6,6).reshape(2,3)
arr.max(axis=0)
arr.max(axis=1)

输出结果

arr:
 [[1. 2. 3.]
 [4. 5. 6.]] 

arr.max(axis=0):
 [4. 5. 6.] 

arr.max(axis=1):
 [3. 6.] 

axis=0 : 跨行操作，计算每列元素
axis=1 : 跨列操作，计算每行元素
arr.shape为（3行，2列），axis=0为行，axis=1为列

三维数组

arr.shape为（2层，3行，2列），axis=0为层，axis=1为行，axis=2为列