如何在NumPy数组中获得N个最大值的索引

在做题过程正遇到一个问题,没有思路·,帮忙解释一下。如何在一个数组中获得N个最大值的索引?

numpy提出了一种通过np.argmax获取数组最大值索引的方法。

我想要一个类似的东西,但是返回N最大值的索引。

例如,如果我有一个数组,[1, 3, 2, 4, 5]和function(array, n=3)将返回对应于元素[5, 4, 3]的索引[4, 3, 1]。

你可以使用np.argsort()来实现,直接取后面N个就好