从 numpy 数组中获取排名

Question

我有一个像下面这样的 numpy 数组values = [0.1, 0.9, 0.8, 0.65, 0.2, 0.7]我需要知道最高价值的价值和指数，我得到了int(np.argmax(values)) max(values)但现在我还需要知道数组中第二高和第三高的值并相应地保留索引。如何修改我的代码以获取这些值？

Answer 1

您可以使用np.argsort，它为您提供最大数字的索引。

indices = np.argsort(values)[::-1]
print(indices)

该[::-1]反转名单，这是必要的，因为argsort回报递增的顺序指数。这给出：

[1, 2, 5, 3, 4, 0]

然后你可以使用

values[indices[n]]

检索第n-th 个最大值。

Answer 2

只需删除最高值，然后使用 int(np.argmax(values))

Answer 3

np.argsort(values)

退货

array([0, 4, 3, 5, 2, 1])