Python如何用while循环提取含指定字符DD的行及其后续内容

原有代码的问题 #

你编写的代码存在几个核心错误：

• 首次进入while循环时line变量未定义，会直接抛出NameError
• 没有设置保留标记位，无法正确判定非=开头的行是否需要保留
• 调用fw.write(templist)会报错，write方法仅支持字符串类型入参，不能直接传入列表
• 拼接非=开头的内容时额外加了逗号、删除了换行，输出格式会和预期不符

正确实现代码 #

核心逻辑是新增keep标记位，用来标记当前读取的内容是否属于需要保留的区块，完全使用while循环实现逐行读取判断：

# 替换为你实际的输入输出路径
input_path = "your_input_file.txt"
output_path = "your_output_file.txt"

with open(input_path, 'r', encoding='utf-8') as fr, open(output_path, 'w', encoding='utf-8') as fw:
    # 标记当前是否属于要保留的内容区块
    keep_content = False
    # 先读取第一行，初始化while循环的判断变量
    current_line = fr.readline()
    while current_line:
        # 遇到以=开头的行，重新判断当前区块是否需要保留
        if current_line.startswith('='):
            if 'DD' in current_line.strip().split():
                keep_content = True
                fw.write(current_line)
            else:
                keep_content = False
        else:
            # 非=开头的行，只要属于保留区块就直接写入
            if keep_content:
                fw.write(current_line)
        # 读取下一行进入下一轮循环
        current_line = fr.readline()

上述代码运行后，输出结果和你给出的期望输出完全一致。

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