numpy.argmax¶
-
numpy.
argmax
(a, axis=None, out=None)[source]¶ 返回沿轴的最大值的索引。
参数: a:array_like
输入数组。
axis:int,可选
默认情况下,索引位于展平的数组中,否则沿着指定的轴。
out:数组,可选
如果提供,结果将被插入到此数组中。它应该是合适的形状和类型。
返回: index_array:intar的ndarray
数组中的数组。它与a.shape具有相同的形状,且沿着轴的尺寸已删除。
笔记
在多次出现最大值的情况下,返回与第一次出现相对应的索引。
例子
>>> 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