argmax(a, axis=None, out=None)
功能:返回沿axis的最大值的索引。
输入参数:
以二维数组为例(对于n维数组,axis可以取0~n-1)
a
:array数组
axis
:①默认情况下,不指定axis时,会把a铺平,然后返回其中最大值的索引;
②axis=0时,返回列对应最大值的索引;
③axis=1时,返回行对应最大值的索引。
返回值:
index_array:整数的ndarray。数组中的索引数组,它的形状与“a.shape”相同,但删除了沿axis的尺寸。
注意:很多推文写的跟我写的相反,即axis=0时取行最大值索引,axis=1时取列最大值索引,我觉得很难理解,例如:这个
官方用例:
a=[[10, 11, 12],
[13, 14, 15]]
print("axis默认时:",np.argmax(a))
print("axis=0时:",np.argmax(a,axis=0))
print("axis=1时:",np.argmax(a,axis=1))
输出结果:
axis默认时: 5
axis=0时: [1 1 1]
axis=1时: [2 2]
注意:当数组维数更大时,我这样理解就会有欠缺,但是对于二维数组是没有问题的。