順序迭代合并后的排序迭代對象

問題

你有一系列排序序列，想將它們合并后得到一個排序序列并在上面迭代遍歷。

解法

heapq.merge() 函數可以幫你解決這個問題。比如：

>>> import heapq >>> a = [1, 4, 7, 10] >>> b = [2, 5, 6, 11] >>> for c in heapq.merge(a, b): ... print(c) ... 1 2 4 5 6 7 10 11

討論

heapq.merge 可迭代特性意味著它不會立馬讀取所有序列。這就意味著你可以在非常長的序列中使用它，而不會有太大的開銷。比如，下面是一個例子來演示如何合并兩個排序文件：

with open('sorted_file_1', 'rt') as file1, \open('sorted_file_2', 'rt') as file2, \open('merged_file', 'wt') as outf:for line in heapq.merge(file1, file2):outf.write(line)

有一點要強調的是 heapq.merge() 需要所有輸入序列必須是排過序的。特別的，它并不會預先讀取所有數據到堆棧中或者預先排序，也不會對輸入做任何的排序檢測。它僅僅是檢查所有序列的開始部分并返回最小的那個，這個過程一直會持續直到所有輸入序列中的元素都被遍歷完。

創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（4.15）：顺序迭代合并后的排序迭代对象的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。