超级会员免费看
本文介绍了在Python中如何使用NumPy的np.argmax/np.argmin获取数组最大值/最小值的索引，并探讨了如何获取数组中前N个最大值/最小值的索引。文中提到了heapq库的nlargest/nsmallest方法，以及使用argsort和argpartition的实现方式，强调heapq方法在获取（index，value）对时的便利性。
优快云 叶庭云：https://yetingyun.blog.youkuaiyun.com/
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=
订阅专栏 解锁全文
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
