Python 删除文件的一行|极客笔记

Python 删除文件的一行

在很多情况下，我们可能需要删除文件中特定的一行内容。Python 提供了多种方法来实现这一目的。在本文中，我们将介绍如何使用 Python 删除文件中的一行内容。

使用 `fileinput` 模块

fileinput 模块可以帮助我们轻松地处理文件的内容。我们可以通过 fileinput 模块的 input 函数来读取文件内容，并且可以指定要删除的行数。

下面是一个示例代码，演示了如何使用 fileinput 模块删除文件中的一行内容：

import fileinput

filename = 'example.txt'
line_to_delete = 3

with fileinput.input(filename, inplace=True) as file:
    for index, line in enumerate(file, start=1):
        if index != line_to_delete:
            print(line, end='')

上述代码中，我们首先导入了 fileinput 模块，然后指定了要操作的文件名和要删除的行号。接着，我们使用 fileinput 模块的 input 函数打开文件，并将 inplace 参数设置为 True，以便在原始文件中直接进行修改。然后，我们遍历文件中的每一行，如果遇到要删除的行号，则跳过该行；否则，将该行内容输出。

使用 `open` 函数和 `writelines` 方法

除了使用 fileinput 模块外，我们还可以使用 open 函数和 writelines 方法来删除文件中的一行内容。这种方法的原理是首先读取文件的所有行，然后将不需要删除的行写入一个新的文件中，最后将新文件覆盖原始文件。

下面是一个示例代码，演示了如何使用 open 函数和 writelines 方法删除文件中的一行内容：

filename = 'example.txt'
line_to_delete = 3

with open(filename, 'r') as file:
    lines = file.readlines()

with open(filename, 'w') as file:
    for index, line in enumerate(lines, start=1):
        if index != line_to_delete:
            file.write(line)

在上述代码中，我们首先使用 open 函数以只读模式打开文件，并使用 readlines 方法将文件内容读入一个列表中。然后，我们再次使用 open 函数以写入模式打开文件，并遍历列表中的行内容，只写入不需要删除的行。