K = 4
a = np.array([0, 8, 0, 4, 5, 8, 8, 0, 4, 2])
# 最大的 k 个元素的索引
print(np.argpartition(a, -K)[-K:]) # [4 1 5 6]
# 最小的 k 个元素的索引
print(np.argpartition(a, K)[:K]) # [7 0 2 9]
python - 输出最大/最小的 k 个元素的索引
最新推荐文章于 2022-07-15 11:38:12 发布