python - 在 numpy 矩阵中找到第 n 个最大值的最快方法

Question

对于单个数组有很多解决方案，但是对于矩阵呢，例如:

>>> k
array([[ 35,  48,  63],
       [ 60,  77,  96],
       [ 91, 112, 135]])

您可以使用 k.max()，但当然这只会返回最高值，135。如果我想要第二个或第三个怎么办？

Answer 1

作为said , np.partition 应该更快(至多 O(n) 运行时间):

np.partition(k.flatten(), -2)[-2]

应该返回第二大元素。 (partition 保证编号的元素就位，前面的元素都小，后面的元素都大)。