Numpy输出矩阵中最大的前5个值所在的索引

#matrix 为矩阵

matrix.flatten()可以将matrix数组平铺变成一维数组，然后利用sort进行排序，[-6]为排序之后排第5的哪个值，where后的matrix > 排第5的值可以输出数组中比第5个值大的值的位置。

import numpy as np

# matrix 为np array
 
max_5_id = np.where(matrix > np.sort(matrix.flatten())[-6])