如何在列表或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])

Answer 2

考虑以下代码， N=5 K = [1,10,2,4,5,5,6,2] #store list in tmp to retrieve index tmp=list(K) #sort list so that largest elements are on the far right K.sort() #To get the 5 largest elements print K[-N:] #To get the 5th largest element print K[-N] #get index of the 5th largest element print tmp.index(K[-N])如果您希望忽略重复项，请按以下方式使用set（）， N=5 K = [1,10,2,4,5,5,6,2] #store list in tmp to retrieve index tmp=list(K) #sort list so that largest elements are on the far right K.sort() #Putting the list to a set removes duplicates K=set(K) #change K back to list since set does not support indexing K=list(K) #To get the 5 largest elements print K[-N:] #To get the 5th largest element print K[-N] #get index of the 5th largest element print tmp.index(K[-N])希望其中之一可以解决您的问题:)

Answer 3

这应该工作：K = [1,2,2,4,5,5,6,10]num = 5print 'K %s.' % (sorted(K, reverse=True)[:num])