如何以numpy返回所有最小索引

  • Z时代
  • 2024-01-10
  • 分类：问答

阅读numpy中argmin函数的文档时，我有些困惑。看起来应该可以完成这项工作：

读这个

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

我可能会认为

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

以上是 如何以numpy返回所有最小索引 的全部内容， 来源链接： utcz.com/qa/419252.html

回到顶部