如何返回numpy中的所有最小索引?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (21)

Numpy中的argmin函数中写到:

Return the indices of the minimum values along an axis.

我可以假设

np.argmin([5, 3, 2, 1, 1, 1, 6, 1])

将返回所有索引的数组:[3, 4, 5, 7]

但是它不是这样的,它只返回3...。我该怎么做才能得到我的结果呢?

提问于
用户回答回答于

多维数组如下:

>>> x = numpy.array([[0, 1],
...                  [3, 2]])
>>> x.argmin(axis=0)
array([0, 0])
>>> x.argmin(axis=1)
array([0, 1])

指定了一个轴,argmin沿着给定的轴获取一维子数组,并返回每个子数组的最小值的第一个索引。它不返回单个最小值的所有索引。

要获得所有最小值的索引,您可以这样做

numpy.where(x == x.min())
用户回答回答于

扫码关注云+社区