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 : 跨列操作，得到列