Python | 快速获取某一列数组中前 N 个最大值/最小值的索引 | 三种方法总结

Posted 2022-08-19 叶庭云

CSDN 叶庭云：https://yetingyun.blog.csdn.net/

---

NumPy 提出了一种通过 np.argmax/np.argmin 获取数组最大值/最小值索引的方法。

但如果想要的是返回 N 个最大值的索引呢？例如，我有一个数组 [1, 4, 5, 3, 2, 8, 6]，那么 nargmax(array, n=3) 将返回与元素 [8, 6, 5] 相对应的索引 [5, 6, 2]，nargmin(array, n=3) 将返回与元素 [1, 2, 3] 相对应的索引 [0, 4, 3]。

Python 实践如下所示：

def get_topk_index(arr, k=3):
    topk_index = heapq.nsmal