Numpy中如何输出Array前N最大值的索引？

我们使用arr.argsort() 函数实现这个功能

top_k=3
arr = np.array([2,5,12,6,1])
top_k_idx=arr.argsort()[::-1][0:top_k]
print(top_k_idx)

接下来我们将之封装为函数:

def ArrayTopK(top_k,arr):
    top_k_idx=arr.argsort()[::-1][0:top_k]
    return top_k_idx