numpy.argmin¶
-
numpy.
argmin
(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.argmin(a) 0 >>> np.argmin(a, axis=0) array([0, 0, 0]) >>> np.argmin(a, axis=1) array([0, 0])
>>> b = np.arange(6) >>> b[4] = 0 >>> b array([0, 1, 2, 3, 0, 5]) >>> np.argmin(b) # Only the first occurrence is returned. 0