擅长:python、mysql、java
<p>你看过内置的numpy<code>argsort</code>方法吗?以下内容:</p>
<p><a href="http://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html" rel="noreferrer">http://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html</a></p>
<p>在我的机器上,我可以用这种方法在29毫秒内对一个有300000个随机浮点数的数组进行排序。</p>
<pre><code>def f(a,N):
return np.argsort(a)[::-1][:N]
</code></pre>