在 python 中按原始顺序查找列表的前 k 个最大项

Question

说我有以下列表：

my_list = [3.5, 1.6, 2.4, 8.9, 5.6]

我想在原来的地方找到前3个最大的数，所以结果应该是：

[3.5, 8.9, 5.6]

我怎么能那样做？ 我想我可以找到最大的 3 个数字并使用过滤器，但我认为比较浮点数可能不是一个好主意。 有什么建议么？

Answer 1

使用堆：

>>> import heapq
>>> heapq.nlargest(3, my_list)
[8.9, 5.6, 3.5]

为相同的想法添加一点润色，以保持它们的原始顺序：

>>> from operator import itemgetter
>>> i_val = heapq.nlargest(3, enumerate(my_list), key=itemgetter(1))
>>> [val for (i, val) in sorted(i_val)]
[3.5, 8.9, 5.6]

Answer 2

您可以按值对索引值对（由enumerate生成）进行排序，获取最后三对，然后按索引对它们进行排序（然后仅从索引值对中获取值，所有这些都在一个-班轮列表理解）：

from operator import itemgetter

my_list = [3.5, 1.6, 2.4, 5.6, 8.9]

result = [p[1] for p in sorted(sorted(enumerate(my_list), key = itemgetter(1))[-3:], key = itemgetter(0))]

print(result)

Output：

[3.5, 5.6, 8.9]

Answer 3

在这里你go，你可以试试这个function：

my_list = [3.5, 1.6, 2.4, 5.6, 8.9]
def select_top(a,array):
    new_list = []
    extra_list = []
    for i in range(len(my_list)):
        extra_list.append(my_list[i])
    final_list = []
    for i in range(a):
        new_list.append(extra_list.index(max(extra_list)))
        extra_list.pop(extra_list.index(max(extra_list)))
    new_list = sorted(new_list,reverse=False)
    for i in new_list:
        final_list.append(array[i])
    return final_list
print(select_top(3,my_list))

我相信它远非最佳，但您可以尽可能多地调整它以获得前 k 个数字并让它们按原始顺序返回。 Output：

[3.5, 5.6, 8.9]

Answer 4

这个怎么样？

[m for m in my_list if m in sorted(my_list)[-3:]]

您正在从前 3 个项目中构建一个新的“m”项目列表。 列表理解使您的项目井井有条。

您的示例的顺序是这样的，您可以对其进行排序并获取前 3 个项目，但我认为您的意思是您可能没有按顺序排列前 3 个项目，例如：

my_list = [3.5, 1.2, 0.3, 7.8, 3.3]

这导致

[3.5,7.8,3.3]

Answer 5

这是我自己的答案，它可以解决问题。

my_list = [3.5, 1.6, 2.4, 8.9, 5.6]
top_k = sorted(my_list)[-3:]
result = list(filter(lambda x : x in top_k,my_list))
print(result)