摘要:問題在某個集合中找到最大或最小的個元素解決方案使用模塊例如此外,這兩個函數(shù)都可以接受作為參數(shù),例如輸出為討論根據(jù)官方文檔對的介紹可以了解到提供了堆數(shù)據(jù)結構的實現(xiàn),并且實現(xiàn)方式是小頂堆,也就是說每次的時候取出的是最小的元素首先使用將一個列
問題
在某個集合中找到最大或最小的N個元素
解決方案使用heapq模塊
heapq.nlargest(n, iterable, key=None)heapq.nsmallest(n, iterable, key=None)
例如:
>>> import heapq >>> l = [9, -2, 0, 8, 1, 3] >>> print(heapq.nlargest(2, l)) [9, 8] >>> print(heapq.nsmallest(2, l)) [-2, 0]
此外,這兩個函數(shù)都可以接受key作為參數(shù),例如:
import heapq fruits = [ {"name": "orange", "price": 5}, {"name": "apple", "price": 2}, {"name": "pear", "price": 1.5}, {"name": "lemon", "price": 3}, ] print(heapq.nlargest(2, fruits, key=lambda x: x["price"]))
輸出為:
[{"name": "orange", "price": 5}, {"name": "lemon", "price": 3}]討論
根據(jù)Python3官方文檔對heapq的介紹可以了解到
heapq提供了堆數(shù)據(jù)結構的實現(xiàn),并且實現(xiàn)方式是小頂堆,也就是說每次pop的時候取出的是最小的元素
首先使用heapq.heapify將一個列表初始化為堆
>>> import heapq >>> l = [-1, 2, 5, 0, 8] >>> heapq.heapify(l) >>> print(l) [-1, 0, 5, 2, 8]
然后就可以調(diào)用heapq.heappush和heapq.heappop對堆進行增加和刪除操作了
>>> heapq.heappush(l, 8) >>> print(l) [-1, 0, 5, 2, 8, 8] >>> print(heapq.heappop(l)) -1
此外,heapq還提供了其他堆的一些操作
heapq.heappushpop(heap, item)先將item存入堆中,然后彈出最小的元素,相當于先調(diào)用了heapq.heappush(item)再調(diào)用heapq.heappop(),但這樣調(diào)用會比分開調(diào)用兩個函數(shù)效率更高
heapq.heapreplace(heap, item)先彈出最小的元素,再存入item
heapq.merge(*iterables, key=None, reverse=False)將多個有序的集合合并成一個有序的集合,并且返回的是迭代器對象
來源Python Cookbook
關注歡迎關注我的微信公眾號:python每日一練
文章版權歸作者所有,未經(jīng)允許請勿轉(zhuǎn)載,若此文章存在違規(guī)行為,您可以聯(lián)系管理員刪除。
轉(zhuǎn)載請注明本文地址:http://specialneedsforspecialkids.com/yun/44679.html
摘要:問題在每日一練中介紹了如何一個可迭代對象,使用運算符即可但往往我們遇到的問題是可迭代對象中的數(shù)量是不確定的這個時候該如何拿到我們想要的元素,比如我們只需要可迭代對象的第一個或者最后一個元素而已解決方案使用中的運算符例如我們需要拿到一個元組的 問題 在每日一練0001中介紹了如何unpack一個可迭代對象,使用,運算符即可 但往往我們遇到的問題是可迭代對象中的數(shù)量是不確定的 這個時候該如...
摘要:問題如何序列化輸出元素包含字符串元組的字符串元組好繞舉個例子將輸出為解決方案容易想到使用函數(shù),但函數(shù)要求元素必須都是字符串類型,否則會拋出錯誤一個比較簡單的方法是將給進中,然后再將給進函數(shù),最后指定函數(shù)的參數(shù)來輸出如果想要將結果存儲起來,那 問題 如何序列化輸出元素包含字符串元組的字符串元組(好繞) 舉個例子 >>> zoo1 = (monkey, elephant) >>> zoo2...
摘要:問題如何執(zhí)行外部命令,如解決方案使用庫在之前,使用函數(shù)在及之后,使用函數(shù)討論命令的執(zhí)行默認不需要環(huán)境,所以當你使用作為參數(shù)時,需要將置位,否則會報錯誤通常來說對于執(zhí)行系統(tǒng)命令,我們會想到,但在官方文檔中已經(jīng)建議了使 問題 如何執(zhí)行外部命令,如ls -l 解決方案 使用subprocess庫 在Python 3.5之前,使用subprocess.call()函數(shù) >>> import s...
摘要:問題如何判斷一個文件是否存在解決方案這個問題可以分成幾類問題如果這里的文件指的是文件或目錄,我們可以用方法如果這里的文件指的是普通的文件,我們可以用方法如果這里的文件指的是目錄,我們可以用方法并且在之后,可以使用面向?qū)ο蟮姆椒ㄊ褂脦靵砼袛啵? 問題 如何判斷一個文件是否存在 解決方案 這個問題可以分成幾類問題 如果這里的文件指的是文件或目錄,我們可以用os.path.exists()方法...
摘要:問題你需要執(zhí)行簡單的日期操作,計算兩個日期間隔多少天某個日期后的多少天是幾月幾日轉(zhuǎn)換時間字符串的格式等解決方案使用庫中的和類其中類代表一個日期時間,例如年月日點分秒類代表一個日期間隔對于實例,可以直接進行數(shù)學運算得到一個實例,也就是兩個日 問題 你需要執(zhí)行簡單的日期操作,計算兩個日期間隔多少天、某個日期后的多少天是幾月幾日、轉(zhuǎn)換時間字符串的格式等 解決方案 使用datetime庫中的d...
閱讀 535·2019-08-30 15:55
閱讀 944·2019-08-29 15:35
閱讀 1199·2019-08-29 13:48
閱讀 1910·2019-08-26 13:29
閱讀 2933·2019-08-23 18:26
閱讀 1237·2019-08-23 18:20
閱讀 2834·2019-08-23 16:43
閱讀 2710·2019-08-23 15:58