python numpy 部分排序 寻找最大的前几个数的方法

作者:猪哥 2018-07-02

如下所示:

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:]]

相关文章

精彩推荐