查找列表的x個最小值的索引

Question

我有一個長度為n的清單。

我想找到保存此列表的5個最小值的索引。

我知道如何使用operator查找包含最小值的索引

min_index,min_value = min(enumerate(list), key=operator.itemgetter(1))

可以更改此代碼以獲得我所追求的5個索引的列表嗎？

Answer 1

盡管這需要對整個列表進行排序，但是您可以從sorted列表中得到一部分：

data = sorted(enumerate(list), key=operator.itemgetter(1))[:5]

Answer 2

那這樣的東西呢？

    map(lambda x: [a.index(x),x],sorted(list)[:5])

這將返回一個列表列表，其中list [x] [0] =索引，list [x] [1] =值

編輯：

假設列表沒有重復的最小值。 正如adhg McDonald-Jensen指出的那樣，這只會返回Give值的第一個實例。

Answer 3

如果使用的軟件包heapq ，它可以這樣做nsamllest ：

heapq.nsmallest(5, enumerate(list), key=operator.itemgetter(1))