np.max、np.min、np.argmax、np.argmin以及np.sum方法均可设置axis参数,设置axis参数后,对应地,函数方法将沿着axis代表的轴的方向进行相应的数学运算。进行数学操作后,其他维度的长度保持不变,axis代表的维度消失。
即:
x = 0 # x表示需要操作的轴的方向,范围为[0,len(a.shape)-1]
np.max(a, axis=x) # 在数组a中沿第x轴求最大值
np.min(a, axis=x) # 在数组a中沿第x轴求最小值
np.argmax(a, axis=x) # 在数组a中沿第x轴求最大值对应的索引
np.argmin(a, axis=x) # 在数组a中沿第x轴求最小值对应的索引
np.sum(a