這篇文章帶給大家的內容是關於python中找到最大或最小的N個元素的實作程式碼，有一定的參考價值，有需要的朋友可以參考一下，希望對你有幫助。
1、需求
我們想在某個集合中找出最大或最小的N個元素
2、解決方案
heapq模組中有兩個函數：nlargest()和nsmallest()
程式碼：
import heapq nums=[1,444,66,77,34,67,2,6,8,2,4,9,556] print(heapq.nlargest(3,nums)) print(heapq.nsmallest(3,nums))
結果：
[556, 444, 77] [1, 2, 2]
這兩個函數都可以接受一個參數key，從而允許他們可以工作在更複雜的資料結構上：
程式碼：
import heapq
portfolio=[
{'name':'IBM','shares':100,'price':91.1},
{'name':'AAPL','shares':50,'price':543.22},
{'name':'FB','shares':200,'price':21.09},
{'name':'HPQ','shares':35,'price':31.75},
{'name':'YHOO','shares':45,'price':16.35},
]
cheap=heapq.nsmallest(3,portfolio,key=lambda s:s['price'])
expensive=heapq.nlargest(3,portfolio,key=lambda s:s['price'])
print(cheap)
print(expensive)結果：
[{'name': 'YHOO', 'shares': 45, 'price': 16.35}, {'name': 'FB', 'shares': 200, 'price': 21.09}, {'name': 'HPQ', 'shares': 35, 'price': 31.75}]
[{'name': 'AAPL', 'shares': 50, 'price': 543.22}, {'name': 'IBM', 'shares': 100, 'price': 91.1}, {'name': 'HPQ', 'shares': 35, 'price': 31.75}]如果只是簡單的找出最小或最大的元素（N=1），那麼使用min()和max()會更快。
以上是在python中找出最大或最小的N個元素的實作程式碼的詳細內容。更多資訊請關注PHP中文網其他相關文章！
