numpy中的argmax()函数

介绍
argmax()：返回某一axis中最大值的索引

参数
numpy.argmax(a, axis=None, out=None)
参数很简单
a：输入的矩阵
axis：指定维数
out：返回的输出矩阵

例子

>>> 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])

n维矩阵最大值的索引

>>> ind = np.unravel_index(np.argmax(a, axis=None), a.shape)
>>> ind
(1, 2)
>>> a[ind]
5

>>> 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

参考：官方文档

哇 不想当搬运工了
numpy系列到此为止