如何无需逐行遍历即可替换文件中指定行的内容？

问题描述 #

有一个大型文件，第9073245行存在问题字符，需要替换/移除该字符，同时保留文件其余内容不变。尝试使用linecache模块实现，但代码仅写入了修改后的目标行，丢失了其他所有内容。

原错误代码：

import linecache
from tempfile import mkstemp
from shutil import move, copymode
from os import fdopen, remove

def lineInFileReplacer(file_path, line_nr, pattern, subst):
    #Create temp file
    fh, abs_path = mkstemp()
    with fdopen(fh,'w') as new_file:
        bad_line = linecache.getline(file_path, line_nr)
        new_file.write(bad_line.replace(pattern, subst))
    #Copy the file permissions from the old file to the new file
    copymode(file_path, abs_path)
    #Remove original file
    remove(file_path)
    #Move new file
    move(abs_path, file_path)

错误原因 #

上述代码仅将修改后的目标行写入临时文件，完全没有处理原文件的其他行，导致最终文件仅保留了修改后的那一行，丢失了所有其他内容。

可行解决方案 #

方案1：流式逐行处理（内存友好） #

这种方式逐行读取文件，仅在目标行执行替换操作，其他行直接写入临时文件，内存占用低，适合大型文件：

from tempfile import mkstemp
from shutil import move, copymode
from os import fdopen, remove

def replace_specific_line(file_path, target_line, pattern, subst):
    # 创建临时文件
    fh, temp_path = mkstemp()
    with fdopen(fh, 'w') as temp_file:
        with open(file_path, 'r') as original_file:
            current_line = 1
            for line in original_file:
                if current_line == target_line:
                    # 替换目标行的指定内容
                    temp_file.write(line.replace(pattern, subst))
                else:
                    # 其他行直接写入
                    temp_file.write(line)
                current_line += 1
    # 复制原文件权限
    copymode(file_path, temp_path)
    # 替换原文件
    remove(file_path)
    move(temp_path, file_path)

方案2：使用fileinput模块（原地修改） #

fileinput模块支持原地修改文件，无需手动处理临时文件，代码更简洁：

import fileinput

def replace_specific_line(file_path, target_line, pattern, subst):
    for line in fileinput.input(file_path, inplace=True):
        if fileinput.lineno() == target_line:
            print(line.replace(pattern, subst), end='')
        else:
            print(line, end='')

注意：inplace=True会自动创建临时文件并替换原文件，使用时建议先备份原文件。

方案3：结合linecache（适合需多次读取指定行的场景） #

如果确实需要使用linecache，必须将原文件所有行都写入临时文件，仅修改目标行：

import linecache
from tempfile import mkstemp
from shutil import move, copymode
from os import fdopen, remove

def replace_specific_line(file_path, target_line, pattern, subst):
    # 获取文件总行数
    with open(file_path, 'r') as f:
        total_lines = sum(1 for _ in f)
    
    # 创建临时文件
    fh, temp_path = mkstemp()
    with fdopen(fh, 'w') as temp_file:
        for line_num in range(1, total_lines + 1):
            line = linecache.getline(file_path, line_num)
            if line_num == target_line:
                temp_file.write(line.replace(pattern, subst))
            else:
                temp_file.write(line)
    # 复制权限并替换原文件
    copymode(file_path, temp_path)
    remove(file_path)
    move(temp_path, file_path)
    # 清理linecache缓存
    linecache.clearcache()

注意：linecache会将文件内容缓存到内存，超大文件可能导致内存占用过高，谨慎使用。

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