在nparray（值和参数）中找到最大的K

Question

我想在一个numpy数组中找到N个最大元素及其参数。

我使用找到的分区，但无法在numpy数组中获得5个最大的元素。这是一个示例：

a =[ 0.409,  0.355, -0.403,  0.307, -0.826, -0.66,  -1.582, -0.112, -0.244, -0.954]    
z = bn.argpartition(-a, 5)[:5]         
z = apodosi.argsort()[-5:]         
z = heapq.nlargest(5, a)   

我会返回数组一段时间，我想获取0,1,3,5,7作为参数，如果可能的话，还可以获取它们的值。我还是编程新手！

Answer 1

最简单的方法可能是这样：

array = [5,3,1,2,4,7,9,8,6]
array.sort(reverse=True) # [9,8,7,...,2,1]
greatest = array[0:5] # [9,8,7,6,5]

我想象numpy数组将以相同的方式工作。