如下所示:
1
2
3
4
|
import numpy as np K = 4 a = np.array([ 0 , 8 , 0 , 4 , 5 , 8 , 8 , 0 , 4 , 2 ]) a[np.argpartition(a, - K)[ - K:]] |
以上这篇python numpy 部分排序 寻找最大的前几个数的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/LeetonFreestyle/article/details/51680127