如何用Python的fileinput正确替换文件指定行？问题解决与替代方案

假设你的test_file.py初始内容为：

line1 = 'abc'
line2 = 'abc'
line3 = 'abc'
line4 = 'abc'

方案一：正确使用fileinput模块 #

你之前出现重复行、行拼接问题，大概率是没正确配置fileinput的参数或处理输出逻辑。以下是正确实现：

import fileinput

# 遍历文件行，inplace=1表示直接修改原文件
for line_num, line in enumerate(fileinput.input('test_file.py', inplace=1), start=1):
    # 去掉行尾换行符避免重复添加
    line = line.rstrip('\n')
    if line_num == 1:
        print("line1 = 'modified_abc_1'")
    elif line_num == 3:
        print("line3 = 'modified_abc_3'")
    else:
        # 其余行原样输出
        print(line)

关键注意点：

• 使用inplace=1参数让fileinput直接修改原文件，无需手动处理临时文件
• 用enumerate指定start=1，让行号从1开始计数（匹配你的lineN命名）
• 先通过rstrip('\n')去除原行的换行符，再用print自动添加标准换行，避免出现行拼接或多余空行

方案二：常规文件读写（无需fileinput） #

如果觉得fileinput不够直观，直接读写文件的方式更易懂，适合小文件场景：

# 读取所有行到内存
with open('test_file.py', 'r', encoding='utf-8') as f:
    lines = f.readlines()

# 修改特定行（注意列表索引从0开始，所以line1对应索引0，line3对应索引2）
if len(lines) >= 1:
    lines[0] = "line1 = 'modified_abc_1'\n"
if len(lines) >= 3:
    lines[2] = "line3 = 'modified_abc_3'\n"

# 写回文件
with open('test_file.py', 'w', encoding='utf-8') as f:
    f.writelines(lines)

注意点：

• 列表索引从0开始，所以要修改第N行，需取索引N-1
• 写回时要确保每行末尾有换行符\n，否则会出现行拼接问题
• 这种方式会把整个文件读入内存，大文件不建议使用，此时fileinput更高效

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