作用:返回向量中的最小值的索引,可以按行返回或者按列返回
1、一维向量
s1=np.array([1,2,3,4])
np.argmin(s1)
>> 0
因为在一维向量s1中,最小值为1,所以返回的索引为0
2、二维向量
s1=np.array([[1,4,3]
,[2,3,4]
,[3,2,1]])
np.argmin(s1,axis=0)
>> [0,2,2]
axis=0按列取最小值,第一列最小值为1,第二列为2,第三列为1,对应的列索引分别为0,2,2,所以返回值为[0,2,2]
np.argmin(s1,axis=1)
>> [0,0,2]
axis=1按行取最小值,第一行最小值为1,第二行为2,第三行为1,对应的索引分别为0,0,2,所以返回值为[0,0,2]