在做题过程正遇到一个问题,没有思路·,帮忙解释一下。如何在一个数组中获得N个最大值的索引?
numpy提出了一种通过np.argmax获取数组最大值索引的方法。
我想要一个类似的东西,但是返回N最大值的索引。
例如,如果我有一个数组,[1, 3, 2, 4, 5]和function(array, n=3)将返回对应于元素[5, 4, 3]的索引[4, 3, 1]。
你可以使用np.argsort()来实现,直接取后面N个就好