順序迭代合并后的排序迭代對(duì)象

問(wèn)題

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

解法

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

>>> 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 可迭代特性意味著它不會(huì)立馬讀取所有序列。這就意味著你可以在非常長(zhǎng)的序列中使用它，而不會(huì)有太大的開(kāi)銷(xiāo)。比如，下面是一個(gè)例子來(lái)演示如何合并兩個(gè)排序文件：

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ǎn)要強(qiáng)調(diào)的是 heapq.merge() 需要所有輸入序列必須是排過(guò)序的。特別的，它并不會(huì)預(yù)先讀取所有數(shù)據(jù)到堆棧中或者預(yù)先排序，也不會(huì)對(duì)輸入做任何的排序檢測(cè)。它僅僅是檢查所有序列的開(kāi)始部分并返回最小的那個(gè)，這個(gè)過(guò)程一直會(huì)持續(xù)直到所有輸入序列中的元素都被遍歷完。

創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎(jiǎng)勵(lì)來(lái)咯，堅(jiān)持創(chuàng)作打卡瓜分現(xiàn)金大獎(jiǎng)

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。