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

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

文章来源: yetingyun.blog.csdn.net，作者：叶庭云，版权归原作者所有，如需转载，请联系作者。

原文链接：yetingyun.blog.csdn.net/article/details/126363942