1、说明
对于较大的数据集,将会占用大量内存。不是对整个组合序列进行排序,而是使用 merge() 一次生成一个新序列。
2、实例
import heapq import random random.seed(2016) data = [] for i in range(4): new_data = list(random.sample(range(1, 101), 5)) new_data.sort() data.append(new_data) for i, d in enumerate(data): print('{}: {}'.format(i, d)) print('\nMerged:') for i in heapq.merge(*data): print(i, end=' ') print() # output # 0: [33, 58, 71, 88, 95] # 1: [10, 11, 17, 38, 91] # 2: [13, 18, 39, 61, 63] # 3: [20, 27, 31, 42, 45] # # Merged: # 10 11 13 17 18 20 27 31 33 38 39 42 45 58 61 63 71 88 91 95
因为merge()使用堆的实现,它根据被合并的序列元素个数消耗内存,而不是所有序列中的元素个数。
以上就是python使用merge实现堆的方法,希望对大家有所帮助。更多Python学习指路:
本文教程操作环境:windows7系统、Python 3.9.1,DELL G3电脑。