找出numpy数组中最大的N个数的索引:

 

def findTopNindex(arr,N):
    return np.argsort(a)[::-1][:N]

测试:

test = np.array([2,5,6,3,4,6,4,8,6,5])
print(findTopNindex(test,3))

>[7 8 5 2]

 

相关文章: