Python未知目标词文本替换代码失效，请求问题排查

用户需求 #

我接触Python及编程仅三周，还请多多包涵。想要实现自定义文本文件转换功能：待处理的文本文件格式固定，但需要替换的目标词是未知的。因此程序需要先找到该目标词，确定匹配模式后，将所有出现的该词替换为指定词汇。

示例文件格式：

For example:
xxxxx
xxxxx
Line3 - word - xxxx xxxx
xxxx xxxx
word
word
xxxx word

Legend:
xxxxx = 每个文件中都存在的固定模板内容
word = 需要替换的随机目标词

目标词的首次出现位置固定，后续会随机出现在文件各处。

原代码及问题 #

原代码如下：

f1 = open('test.txt', 'r')
f2 = open('file2.txt', 'w')

pattern = ''
for line in f1.readlines():
    if line.startswith('Seat 1'):
        line = line.split(' ', 3)
        pattern = line[2]
        line = ' '.join(line)
        f2.write(line)
    elif pattern in line.strip():
        f2.write(line.replace(pattern, 'NewWord'))
    else:
        f2.write(line)
f1.close()
f2.close()

代码无法正常运行的核心问题：

1. 初始pattern为空字符串：空字符串会被判定为存在于任何字符串中，导致程序一开始就执行替换逻辑。line.replace('', 'NewWord')会在每个字符之间插入NewWord，完全破坏原文件内容。
2. split(' ', 3)的不可靠性：如果Seat 1开头的行包含多个连续空格或制表符，split(' ',3)会把连续空格当成多个分割符，导致提取的pattern错误；同时分割后再join会把原行的多空格变成单空格，破坏原格式。
3. 缺少异常防护：如果Seat 1行的格式不符合预期（比如不足3个分割部分），直接取line[2]会触发索引越界报错。
4. 文件操作不够安全：手动调用close()如果遇到程序异常，可能导致文件未正常关闭。

修复后的代码 #

# 用上下文管理器自动处理文件的打开与关闭，避免资源泄漏
with open('test.txt', 'r') as f1, open('file2.txt', 'w') as f2:
    pattern = None
    for line in f1:
        # 定位目标词所在的固定行
        if line.startswith('Seat 1'):
            # maxsplit=3 表示最多分割3次，自动处理任意空白符（多空格、制表符）
            parts = line.split(maxsplit=3)
            # 确保分割后有足够的元素，避免索引越界
            if len(parts) >= 3:
                pattern = parts[2]
            # 原行直接写入，不修改这一行的内容
            f2.write(line)
        elif pattern is not None:
            # 仅当找到目标词后，才执行替换操作
            f2.write(line.replace(pattern, 'NewWord'))
        else:
            # 未找到目标词前，直接写入原行
            f2.write(line)

关键改进说明 #

• 将pattern初始化为None，只有在成功提取目标词后才启动替换逻辑，彻底避免初始阶段的错误替换。
• 使用split(maxsplit=3)替代原有的分割方式，适配各种空白符分隔的情况，同时保留原行的空格格式。
• 增加分割后元素长度的判断，防止因行格式异常导致程序崩溃。
• 采用with语句管理文件，无需手动关闭，代码更简洁安全。

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