numpy.argmin

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

译者:飞龙 UsyiyiCN

校对:(虚位以待)

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

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

参数:

a:array_like

输入数组。

axis:int,可选

默认情况下,索引位于展平的数组中,否则沿着指定的轴。

out:数组,可选

如果提供,结果将被插入到此数组中。它应该是合适的形状和类型。

返回:

index_array:intar的ndarray

数组中的数组。它与a.shape具有相同的形状,且沿着的尺寸已删除。

也可以看看

ndarray.argminargmax

amin
沿给定轴的最小值。
unravel_index
将平坦索引转换为索引元组。

笔记

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

例子

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