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

Question

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

Answer 1

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

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

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