我们使用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
我们使用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