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 部分排序 寻找最大的前几个数
最新推荐文章于 2025-06-30 09:04:04 发布

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