Быстрый способ найти самые большие N элементов в массиве numpy

Я знаю, что могу сделать это следующим образом:

import numpy as np
N=10
a=np.arange(1,100,1)
np.argsort()[-N:]

Однако это очень медленно, так как он сделал полную сортировку.

Интересно, предоставляет ли numpy какие-то методы, чтобы сделать это быстро.

43
задан Hailiang Zhang 26 April 2012 в 16:29
поделиться