numpy argmin 用法及代码示例

用法:
numpy.argmin(a, axis=None, out=None)
返回沿轴的最小值的索引。

描述
参数 a: : array_like

输入数组。
axis: : int, 可选参数

默认情况下,索引位于平面数组中,否则沿指定的轴。
out: : array, 可选参数

如果提供,结果将被插入此数组。它应该具有适当的形状和dtype。

返回值 index_array: : 整数数组

索引数组到数组中。它的形状与a.shape相同,但沿轴的尺寸已删除。

注意:

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

例子:

>>> a = np.arange(6).reshape(2,3) + 10
>>> a
array([[10, 11, 12],
       [13, 14, 15]])
>>> np.argmin(a)
0
>>> np.argmin(a, axis=0)
array([0, 0, 0])
>>> np.argmin(a, axis=1)
array([0, 0])

N-dimensional数组的最小元素的索引:

>>> ind = np.unravel_index(np.argmin(a, axis=None), a.shape)
>>> ind
(0, 0)
>>> a[ind]
10
>>> b = np.arange(6) + 10
>>> b[4] = 10
>>> b
array([10, 11, 12, 13, 10, 15])
>>> np.argmin(b)  # Only the first occurrence is returned.
0

发表回复

登录... 后才能评论