簡體   English   中英

在列表中找到第n個最小的數字?

[英]Finding the nth smallest number in a list?

我需要一種有效的方法來獲取第n個最小的數字及其索引,該列表包含多達15000個實體(因此速度並不是至關重要的)。

可悲的是我不能使用numpy或任何其他非標准庫。

我正在使用Python 2.7

使用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)]

使用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)]

position8value30

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM