提出問題：

如何在某集合里面找出最大或最小的K個元素。

解決思路：

找出最大或最下的K個元素，可以使用Python庫中的heapq模塊，該模塊提供兩個函數(shù)nlargest()求最大K個和nsmallest()求最小K個。

下面我們舉例說明：

import heapq

nums=[12,-9,-3,32,9,56,23,0,11,34]
print(heapq.nlargest(4,nums)) #-->最大的4個
print(heapq.nsmallest(4,nums)) #-->最小的4個

運行結果：

[56, 34, 32, 23]
[-9, -3, 0, 9]

分析下，nlargest()和nsmallest()函數(shù)有兩個參數(shù)，第一個參數(shù)是求最大或最下的K個元素，第二個參數(shù)是待查詢的集合。除此之外，他們也可以接受一個參數(shù)key，這使得他們處理更加復雜的數(shù)據(jù)結構。例如：

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},
{'name': 'ACME', 'shares': 75, 'price': 115.65}
]

cheap = heapq.nsmallest(3, portfolio, key=lambda s: s['price'])
print(cheap)
expensive = heapq.nlargest(3, portfolio, key=lambda s: s['price'])
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': 'ACME', 'shares': 75, 'price': 115.65}, {'name': 'IBM', 'shares': 100, 'price': 91.1}]

深入討論：

假如說，我們正在尋找某集合中最大或最下的K個元素，并且N的數(shù)值很小，如果再使用上面的方法，可能就不是最好的選擇。那么，我們介紹heapify()函數(shù)，這個函數(shù)首先會在底層將數(shù)據(jù)轉換成列表，并且元素會以堆的順序排列。舉例：

從上面的代碼中，我們看出元素確實是以堆的順序排列的，同時堆hea_num[0]對應的元素-9總是最小的。在heapq（）模塊中還提供heappop()函數(shù)，該方法會把第一個元素（最小的）給彈出來，然后第二小的元素會自動補位，它的操作時間復雜度是O(log N),其中N代表的是堆的大小。

具體的操作看下面的代碼：

總結一下：

當要查找的元素數(shù)量比較少的時，適合使用nlargest()和nsmallest()

當只查找集合中最大或最小的1個元素時，推薦使用min()和max()

當N和集合本身大小差不多時，應該是先對集合排序，然后做切片操作（比如：sorted(items)[:N]或sorted(items)[-N:]）

補充知識：python三個數(shù)從小到大排序

python三個數(shù)從小到大排序

1、首先定義一個函數(shù)paiLie()；然后在paiLie函數(shù)內(nèi)使用for循環(huán)和input獲取三個數(shù)字并存入列表；最后調(diào)用列表的sort()方法進行排序即可。

def paiLie():
 result = []
 for i in range(3):
  x = input("請輸入數(shù)字:")
  result.append(x)
 result.sort()
 print result

2、調(diào)用

paiLie()
請輸入數(shù)字:56
請輸入數(shù)字:5
請輸入數(shù)字:89

運行結果:

[5, 56, 89]

以上這篇Python實現(xiàn)從N個數(shù)中找到最大的K個數(shù)就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

最新評論