numpy.argmax

作用：返回轴上最大值的索引。

例子：

a = np.arange(6).reshape(2,3)
a
array([[0, 1, 2],
       [3, 4, 5]])
np.argmax(a)
5
np.argmax(a, axis=0)#列上最大值的索引
array([1, 1, 1])
 np.argmax(a, axis=1)#行上最大值的索引
array([2, 2])

例子2：

• 用来返回最大值在N维数组中的坐标索引：

ind = np.unravel_index(np.argmax(a, axis=None), a.shape)#np.unravel_index将平面索引或平面索引数组转换为坐标数组
ind
(1, 2)
a[ind]
5

例子3：

• 只有第一个最大值的索引被返回。

b = np.arange(6)
b[1] = 5
b
array([0, 5, 2, 3, 4, 5])
np.argmax(b)  # Only the first occurrence is returned.
1