python - 如何找到列表或 np.array 中 n 个最大元素的索引，Python

Question

是否有内置函数或非常简单的方法来查找列表或 numpy 数组中 n 个最大元素的索引？

K = [1,2,2,4,5,5,6,10]

找到最大的 5 个元素的索引？

我多次计算重复项，输出应该是那些最大数字的索引列表

Answer 1

也许是这样的:

>>> K
[4, 5, 1, 6, 2, 5, 2, 10]
>>> sorted(range(len(K)), key=lambda x: K[x])
[2, 4, 6, 0, 1, 5, 3, 7]
>>> sorted(range(len(K)), key=lambda x: K[x])[-5:]
[0, 1, 5, 3, 7]

或者使用numpy，你可以使用argsort:

>>> np.argsort(K)[-5:]
array([0, 1, 5, 3, 7])

argsort也是一个方法:

>>> K = np.array(K)
>>> K.argsort()[-5:]
array([0, 1, 5, 3, 7])
>>> K[K.argsort()[-5:]]
array([ 4,  5,  5,  6, 10])