Numpy 最大值与最大值索引函数

numpy.amin() 和 numpy.amax()

#numpy.amin() 和 numpy.amax() 这些函数从给定数组中的元素沿指定轴返回最小值和最大值。

import numpy as np
a=np.array([[3,7,5],[8,4,3],[2,4,9]])
print(np.amax(a)) #输出为9 没有指定轴会将数组展开
print(np.amax(a,axis=0))#[8 7 9] 寻找每一列最大值
print(np.amax(a,axis=1))#[7 8 9] 寻找每一行最大值

numpy.argmax() 和 numpy.argmin()

numpy.argmax() 和 numpy.argmin() 这两个函数分别沿给定轴返回最大和最小元素的索引。

a = np.array([[30,40,70],[80,20,10],[50,90,60]])
print(np.argmax(a))#输出90的索引7
print(np.argmax(a,axis=0))#[1 2 0]
print(np.argmax(a,axis=1))#[2 0 1]