Python 删除指定行|极客教程

Python 删除指定行

在处理文本文件时，有时候我们需要删除文件中的某些指定行。Python 提供了多种方法来实现这个目标。在本文中，我们将讨论如何使用 Python 来删除文本文件中的指定行。

使用文件的读写方法

一种简单的方法是将文件的内容读入内存中，然后只保留我们需要的行，最后将处理后的内容写回到文件中。

def delete_lines(file_path, lines_to_delete):
    with open(file_path, 'r') as f:
        lines = f.readlines()

    with open(file_path, 'w') as f:
        for index, line in enumerate(lines):
            if index + 1 not in lines_to_delete:
                f.write(line)

# 测试
file_path = 'test.txt'
lines_to_delete = [2, 4]
delete_lines(file_path, lines_to_delete)

上面的代码中，我们定义了一个 delete_lines 函数，它接受两个参数，分别是文件路径 file_path 和需要删除的行号列表 lines_to_delete。函数首先打开文件，将文件的内容读入 lines 列表中，然后遍历 lines 列表，只保留行号不在 lines_to_delete 列表中的行，并将结果写回文件中。

使用临时文件

另一种方法是创建一个临时文件，将需要保留的行写入临时文件中，最后将临时文件重命名为原文件名。

import os

def delete_lines(file_path, lines_to_delete):
    temp_file = 'temp.txt'

    with open(file_path, 'r') as f:
        with open(temp_file, 'w') as temp:
            for index, line in enumerate(f):
                if index + 1 not in lines_to_delete:
                    temp.write(line)

    os.remove(file_path)
    os.rename(temp_file, file_path)

# 测试
file_path = 'test.txt'
lines_to_delete = [2, 4]
delete_lines(file_path, lines_to_delete)

上面的代码中，我们定义了一个 delete_lines 函数，它与之前的方法类似，不同之处在于我们使用了临时文件来保存结果。在函数中，我们打开原文件和临时文件，在遍历原文件内容时，只保留需要保留的行，并将结果写入临时文件中。最后，我们删除原文件并将临时文件重命名为原文件名，从而完成删除操作。

使用文件迭代器

如果文件较大，一次性读取整个文件可能会导致内存溢出。此时，可以使用文件迭代器来处理文件，逐行读取和写入，从而避免占用过多内存。

import os

def delete_lines(file_path, lines_to_delete):
    temp_file = 'temp.txt'

    with open(file_path, 'r') as f:
        with open(temp_file, 'w') as temp:
            for index, line in enumerate(f, start=1):
                if index not in lines_to_delete:
                    temp.write(line)

    os.remove(file_path)
    os.rename(temp_file, file_path)

# 测试
file_path = 'test.txt'
lines_to_delete = [2, 4]
delete_lines(file_path, lines_to_delete)

在上面的代码中，我们使用了文件迭代器 enumerate(f, start=1) 来遍历文件内容。文件迭代器可以逐行读取文件，并返回行号和行内容。这样我们就可以避免一次性读取整个文件，从而节省内存。