如果只是簡單的想找到最小或最大的元素（n=1時），使用min和max最快
如果n和集合的本身大小差不多，通常更快的方法是先對集合進行排序，然後做切片操作，例如sorted(items)[:n]或者sorted(items)[-n:]
如果我們想在某個集合中找到最大或者最小的n個元素,使用heapq中的nlargest和nsmallest，這兩個函式的實際實現會該劇使用它們的方式不同而有所不同，可能會相應的做出一些優化措施，比如當n接近集合大小的時候，採用先排序的方法
import heapq
nums = [1, 2, 3, 423, 54, 3232, -4, -554, 99932, 555]
print(heapq.nlargest(2, nums))
print(heapq.nsmallest(2, nums))
# 這兩個函式都可以使用key引數來處理複雜的資料結構
protfolio = [,,
,,,,
]cheap = heapq.nsmallest(3, protfolio, key=lambda x: x.get('price'))
expensive = heapq.nlargest(3, protfolio, key=lambda x: x.get('price'))
print(cheap)
print(expensive)
[99932, 3232]
[-554, -4]
[, , ]
[, , ]
如果n很小，那麼使用堆排序可能會效能更好
import heapq
nums = [1, 2, 3, 423, 54, 3232, -4, -554, 99932, 555]
heapq.heapify(nums)
print(nums)
import heapq
nums = [1, 2, 3, 423, 54, 3232, -4, -554, 99932, 555]
heapq.heapify(nums)
print(nums)
[-554, 1, -4, 2, 54, 3232, 3, 423, 99932, 555]
-554-41
找到最大或最小的N個元素
在某個集合中找出 最大或最小的n個元素 heapq模組的nsmallest 和 nlargest是我們所需要的。如果只是找最大和最小max min更適合我們 更快 import heapq portfolio 可以接受key值 cheap heapq.nsmallest 3,portfolio,ke...
N個元素中選最大最小
n 個元素中選最大最小 問題描述 從n 個元素中，選出最大元素和最小元素。分析複雜度。方法 方法有三種。方法一 最笨的方法，迴圈一次找出最大的算素。再迴圈一下找到最小的。這裡2 次迴圈可以放在一起，作為小小的優化。比較次數 2 n 1 2n 2。void select max min 1 int a...
Python查詢最大或最小的 N 個元素
本文為python cookbook第三版1.4的內容 怎樣從乙個集合中獲得最大或者最小的 n 個元素列表？heapq 模組有兩個函式 nlargest 和nsmallest 可以完美解決這個問題。import heapq nums 1,8,2,23,7,4,18,23,42,37,2 print ...