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