python - 我需要一个 numpy 数组中的 N 个最小(索引)值

Question

嗨，我有一个包含 X 个值的数组，我想找到十个最小值的索引。在这个链接中，他们有效地计算了最大值，How to get indices of N maximum values in a numpy array?但是我还不能对链接发表评论，所以我不得不重新发布这个问题。

我不确定需要更改哪些指标才能达到最小值而不是最大值。这是他们的代码

In [1]: import numpy as np

In [2]: arr = np.array([1, 3, 2, 4, 5])

In [3]: arr.argsort()[-3:][::-1]
Out[3]: array([4, 3, 1]) 

Answer 1

如果你打电话

arr.argsort()[:3]

它将为您提供 3 个最小元素的索引。

array([0, 2, 1], dtype=int64)

所以，对于 n，你应该调用

arr.argsort()[:n]