在列表中找到第 n 个最小的数字？

Posted 2023-02-23

【中文标题】在列表中找到第 n 个最小的数字？

【英文标题】：Finding the nth smallest number in a list?

【发布时间】：2014-02-21 22:15:54

【问题描述】：

我需要一种有效的方法来获取第 n 个最小的数字及其在包含多达 15000 个实体的列表中的索引（因此速度不是超级关键）。

很遗憾，我不能使用 numpy 或任何其他非标准库。

我正在使用 Python 2.7

【问题讨论】：

但对列表进行排序会丢失其原始索引

@mhlester：如果你使用sorted则不会@

【参考方案1】：

使用heapq.nsmallest（和enumerate获取索引）：

nums = [random.randint(1,1000000) for _ in range(10000)]

import heapq
import operator

heapq.nsmallest(10,enumerate(nums),key=operator.itemgetter(1))
Out[26]: 
[(5544, 35),
 (1702, 43),
 (6547, 227),
 (1540, 253),
 (4919, 360),
 (7993, 445),
 (1608, 495),
 (5832, 505),
 (1388, 716),
 (5103, 814)]

【讨论】：

对于由 10、10、10、9、8 组成的列表 - 可以说第三小的可能是 8 而不是 10

我想这是可能的，但我认为这是不太直观的解释。如果 OP 需要重复数据删除，我会解决它。

【参考方案2】：

使用enumerate存储原始位置，然后按值排序：

original_list = [36, 44, 23, 24, 47, 19, 49, 36, 30, 3]
sorted_lookup = sorted(enumerate(original_list), key=lambda i:i[1])
position, value = sorted_lookup[4] # 4 is my "n"

在我的示例中，sorted_lookup 是：

[(9, 3), (5, 19), (2, 23), (3, 24), (8, 30), (0, 36), (7, 36), (1, 44), (4, 47), (6, 49)]

而position 是8，value 是30