如何使用Python删除TXT文件中包含特定字符串的行之后的所有内容？

现有代码问题说明 #

你当前的代码存在几个可优化/修复的点：

• 首次遍历读取文件后，文件指针已移动到文件末尾，后续it.islice操作无法读取到任何内容，最终生成的新文件会是空文件
• 未处理目标字符串不存在的场景，此时变量num未定义，运行时会抛出NameError异常
• 不需要分两次读取文件，单次遍历即可完成需求，内存占用和运行效率都更优

优化实现方案 #

方案1：单次遍历直接截断（推荐，适合绝大多数场景） #

不需要提前计算行号，遍历文件时边读边写，找到目标行后直接终止读取即可：

target_str = "string"

with open('sampletext.txt', "r", encoding="utf-8") as rf, open('sampletext_copy.txt', "w", encoding="utf-8") as wf:
    for line in rf:
        wf.write(line)
        if target_str in line:
            # 找到目标行后直接跳出循环，不再处理后续内容
            break
    else:
        # 循环未触发break，说明文件中不存在目标字符串
        print(f"未找到包含指定字符串「{target_str}」的行，已保留原文件全部内容")

方案2：先获取行号再截断（适用于需要提前知道行号的场景） #

如果你的业务场景确实需要先拿到目标行的行号做其他处理，记得在二次读取前重置文件指针：

import itertools as it

target_str = "string"
target_line_num = None

with open('sampletext.txt', "r", encoding="utf-8") as rf:
    # 查找目标行号
    for num, line in enumerate(rf, 1):
        if target_str in line:
            target_line_num = num
            print(f"目标字符串所在行号：{target_line_num}")
            break
    else:
        print(f"未找到包含指定字符串「{target_str}」的行")
        # 可根据需求调整逻辑，比如设置target_line_num为总行数保留全部内容，或直接退出

    if target_line_num is not None:
        # 重置文件指针到文件开头
        rf.seek(0)
        with open('sampletext_copy.txt', "w", encoding="utf-8") as wf:
            for line in it.islice(rf, target_line_num):
                wf.write(line)

额外注意事项 #

• 处理大文件时优先选择方案1，不需要加载整个文件到内存，也不需要二次遍历，性能最优
• 上述代码默认取目标字符串第一次出现的位置做截断，若需要匹配最后一次出现的位置，可调整匹配逻辑暂存最后一次的行号，再做截断
• 如果需要原地修改原文件而非生成副本，建议先将内容写入临时文件，验证处理结果无误后再替换原文件，避免修改失败损坏原文件

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