numpy.argmin

原型

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

功能

给出axis方向最小值的下标

Parameters:

• a : Input array.

• axis : 默认将输入数组展平。否则，按照axis方向

• out : 可选

Returns:

index_array : 下标组成的数组。shape与输入数组a去掉axis的维度相同。

举例：

1、展平、axis=0、axis=1

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

2、多个最小值，只显示第一个

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

————————————————
原文链接：https://blog.youkuaiyun.com/weixin_41770169/article/details/80714461