numpy argmin 用法及代码示例
用法:
numpy.argmin(a, axis=None, out=None)
返回沿轴的最小值的索引。
– | 描述 |
---|---|
参数 | a: : array_like
输入数组。 默认情况下,索引位于平面数组中,否则沿指定的轴。 如果提供,结果将被插入此数组。它应该具有适当的形状和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
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。