如何用Python根据条件在文本文件中添加换行符？

原代码存在的问题 #

• 仅处理第一个匹配项：find()方法只会返回NODATACODE第一次出现的索引，无法处理文件中多次出现的情况
• 丢失前置内容：data[nodata_index:]只保留了第一个匹配点之后的文本，原始文件中匹配点之前的内容全部被丢弃
• 错误的换行处理逻辑：split('\n')将文本拆分为列表后再转成字符串，会输出类似['xxx', 'xxx']的格式，完全不符合预期的换行效果

修改后的代码 #

with open('InputFile.txt', 'r') as file:
    data = file.read()

# 全局替换所有'NODATACODE'，在每个匹配字符串后添加换行符
modified_data = data.replace('NODATACODE', 'NODATACODE\n')

with open('Output.txt', 'w') as file:
    file.write(modified_data)

# 可选：提示替换结果
if 'NODATACODE' in data:
    print("已完成替换，结果已写入Output.txt")
else:
    print("'NODATACODE' not found in the file.")

代码说明 #

• 全局处理所有匹配：replace()方法默认会替换文本中所有出现的NODATACODE，无需逐个查找索引
• 保留完整原始内容：直接对整个文件内容进行修改，不会丢失任何部分
• 精准实现需求：直接在每个NODATACODE后添加换行符，输出格式完全符合预期

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