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

2018-02-24 15:26 更新

問題

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

解決方案

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

>>> 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)

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

以上內(nèi)容是否對您有幫助：
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號