numpy.argmax

原文：https://docs.scipy.org/doc/numpy/reference/generated/numpy.argmax.html

译者：飞龙 UsyiyiCN

校对：（虚位以待）

numpy.argmax(a, axis=None, out=None)[source]

返回沿轴的最大值的索引。

参数：	a：array_like 输入数组。 axis：int，可选 默认情况下，索引位于展平的数组中，否则沿着指定的轴。 out：数组，可选 如果提供，结果将被插入到此数组中。它应该是合适的形状和类型。
返回：	index_array：intar的ndarray 数组中的数组。它与a.shape具有相同的形状，且沿着轴的尺寸已删除。

也可以看看

ndarray.argmax，argmin

amax

沿给定轴的最大值。

unravel_index

将平坦索引转换为索引元组。

笔记

在多次出现最大值的情况下，返回与第一次出现相对应的索引。

例子

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

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