沿NumPy数组中的轴获取N个最大值和索引

沿NumPy数组中的轴获取N个最大值和索引

我会用

argsort()

：

top2_ind = score_matrix.argsort()[:,::-1][:,:2]

也就是说，产生一个包含索引的数组，该索引将进行排序

score_matrix

：

array([[1, 2, 0],       [0, 1, 2],       [0, 1, 2]])

然后使用反转列

::-1

，然后使用

:2

：提取前两列：

array([[0, 2],       [2, 1],       [2, 1]])

然后类似，但定期

np.sort()

获取值：

top2_score = np.sort(score_matrix)[:,::-1][:,:2]

遵循上述相同的机制，可以为您提供：

array([[ 1. ,  0.4],       [ 0.8,  0.6],       [ 0.5,  0.3]])