python - 使用生成器按时间优先读取两个文件

Question

我刚刚阅读了生成器，想知道您将如何使用生成器：
假设有两个文件，每个文件都有一个时间列。每个文件都按时间升序排序，我希望使用时间优先级来抓取这些文件中的行。而不是写一个简单的表达式（见下文），我想知道创建一个生成器 next() 是否是按时间优先级读取这两个文件的更好/合适的方法。

for line1 in file1:    
    do_something
    try:
        if time1<time2:
            do_something    
            continue
        else:
            do_something  
    except:
        pass  
    for line2 in file2:
        do_something  
        if time2>time1:
            break  

Answer 1

使用heapq.merge

def generate_timeline(file):
    for line in file:
         time1 = extract_time_from_line(line)
         yield time1, line


for (time1, line) in heapq.merge(generate_timeline(file1), generate_timeline(file2)):
    process(line)