双文件迭代优先级处理需求及代码修正咨询

问题分析 #

你需要实现的核心逻辑是：

• 优先处理file1中连续的#开头行，直到连续块结束
• 切换到file2，从上次暂停的位置继续读取输出
• 两个文件的读取位置各自独立，操作一个时另一个位置保持不变

之前用zip/zip_longest的方式无法满足需求，因为这类方法会同步推进两个迭代器的位置，无法实现“优先处理某一个的连续块”的逻辑。

解决方案代码 #

下面是基于迭代器特性实现的Python代码，完全符合需求：

def process_files(file1, file2):
    # 把文件对象转为惰性迭代器，各自维护读取位置
    iter1 = iter(file1)
    iter2 = iter(file2)
    # 缓存file1中读到的非#行（迭代器无法回退，用缓存保存未处理的行）
    cache1 = None

    while True:
        # 阶段1：优先处理file1的连续#行块
        comment_block = []
        # 先检查缓存中的行
        current_line = cache1
        if current_line is not None:
            if current_line.strip().startswith('#'):
                comment_block.append(current_line)
                current_line = None
            else:
                # 缓存行不是注释，跳过阶段1
                pass

        # 继续从file1读取连续注释行
        if current_line is None:
            try:
                while True:
                    line = next(iter1)
                    if line.strip().startswith('#'):
                        comment_block.append(line)
                    else:
                        # 遇到非注释行，存入缓存，停止读取
                        cache1 = line
                        break
            except StopIteration:
                # file1已读完，清空缓存
                cache1 = None

        # 如果收集到注释块，输出并回到循环开头继续检查
        if comment_block:
            for line in comment_block:
                print(line, end='')
            continue

        # 阶段2：处理file2的行
        try:
            line = next(iter2)
            print(line, end='')
        except StopIteration:
            # file2已读完，处理file1剩余所有内容
            if cache1 is not None:
                print(cache1, end='')
                cache1 = None
            # 输出file1剩下的所有行
            for line in iter1:
                print(line, end='')
            # 所有内容处理完毕，退出循环
            break

使用示例 #

假设你有两个文件：
file1.txt内容：

# Header 1
# Header 2
Regular line 1
# Header 3
Regular line 2

file2.txt内容：

Content line 1
Content line 2
Content line 3

调用方式：

with open('file1.txt', 'r') as f1, open('file2.txt', 'r') as f2:
    process_files(f1, f2)

输出结果（符合期望） #

# Header 1
# Header 2
Content line 1
Content line 2
Content line 3
Regular line 1
# Header 3
Regular line 2

逻辑说明 #

1. 迭代器独立性：iter1和iter2是独立的惰性迭代器，只有调用next()时才会推进读取位置，未操作时位置保持不变
2. 注释块处理：优先扫描file1的连续#行，用cache1保存遇到的第一个非注释行（避免迭代器回退的问题）
3. 切换逻辑：处理完注释块后，自动切换到file2读取一行；当file2耗尽时，一次性输出file1剩余的所有内容

内容的提问来源于stack exchange，提问作者Gecko