numpy.argmax()函数参数解释

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]

注意：当数组维数更大时，我这样理解就会有欠缺，但是对于二维数组是没有问题的。