列表的 N 个最小元素的索引

Question

我试图从一个列表中找到 3 个最低的数字，并使用这些索引从另一个列表中找到相应的值。 这是我尝试过的一个例子

a = [12, 83, 22, 30, 57, 32, 88, 46, 20, 26, 78, 65, 45, 56, 74]

b = ['a','b','c','d','e','f','g','h','i','j','k','l','m','n','o']

lowest = lowest2 = lowest3 =  max(a)

indexes = []

for i in range(0,len(a)):
    if a[i] < lowest :  
            lowest = a[i] 
            print(b[i])

    elif a[i] < lowest2 and a[i] > lowest:  
            lowest2 = a[i]
            print(b[i])

    elif a[i] < lowest3 and a[i] > lowest2:  
            lowest3 = a[i]
            print(b[i])

    print(lowest,lowest2,lowest3)

我只能使用 anaconda 库，不，这不是分配，这是我一直在尝试做的程序的一小部分。

输出：一个 b C d 一世 j 12 20 26

Answer 1

你可以做的是对列表a进行排序并返回它的索引值，

>>> a = [12, 83, 22, 30, 57, 32, 88, 46, 20, 26, 78, 65, 45, 56, 74]
>>> n_min_values = 3 
>>> sorted(range(len(a)), key=lambda k: a[k])[:n_min_values] 
[0, 8, 2] # index values

并遍历此索引值列表以从列表b中获取相应的值，

>>> b = ['a','b','c','d','e','f','g','h','i','j','k','l','m','n','o']
>>> for i in sorted(range(len(a)), key=lambda k: a[k])[:n_min_values ]:
        print(b[i])
a
i
c

或

使用列表理解，

>>> [b[i] for i in sorted(range(len(a)), key=lambda k: a[k])[:n_min_values ]]
['a', 'i', 'c'] # your output

Answer 2

堆擅长做这种事情，尤其是在最小项目的顺序不重要的情况下。

这是一个在 a 上使用 heapq.nsmallest 及其压缩索引的小示例。

from heapq import nsmallest
from operator import itemgetter

n = 3
idx, _ = zip(*nsmallest(n, enumerate(a), key=itemgetter(1)))
idx
# (0, 8, 2)

[b[i] for i in idx]  # your result
# ['a', 'i', 'c']