一文掌握Python 中的文件处理

liftword4天前技术文章1

了解 Python 中的文件操作

打开文件：文件处理的基础

Python 中的 open（）函数是文件处理的基础。它打开一个文件并返回一个 file 对象，允许我们与该文件进行交互。打开文件的模式决定了我们可以执行的操作。以下是主要模式：

读取（'r'）：打开文件进行读取。这是默认模式，文件必须存在。
Write （'w'）：打开文件进行写入。如果文件存在，则将其截断。如果不存在，则创建一个新文件。
Append （'a'）：打开文件进行写入，但附加到文件末尾，而不是截断文件。
读/写（'r+'， 'w+'， 'a+'）：允许读取和写入操作。

示例：创建和写入文件

with open('example.txt', 'w') as file:
    file.write('Advanced File Handling in Python')

在此示例中，在写入模式下打开（或创建） example.txt 并向其写入字符串。with 语句可确保在操作后正确关闭文件。

读取文件：高效检索数据

Python 提供了多种读取文件的方法，允许根据文件大小和所需的操作来定制我们的方法：

read（）：将整个文件作为单个字符串读取。
readline（）：一次读取一行。
readlines（）：将整个文件读入行列表。

示例：逐行读取文件

with open('example.txt', 'r') as file:
    for line in file:
        print(line.strip())

在此示例中，我们在读取模式下打开文件并遍历每一行，在删除任何尾随空格后打印它。

写入文件：安全地存储数据

在 Python 中写入文件很简单，但需要小心处理以避免数据丢失。write（）方法是此操作的主要方法。

示例：将数据附加到文件

with open('example.txt', 'a') as file:
    file.write('\nAppending new content to the file.')

在这里，我们以追加模式打开文件并添加新文本行，同时保留现有内容。

关闭文件：确保数据完整性

在执行操作后关闭文件对于确保所有数据都写入磁盘并释放资源至关重要。Python 中的 with 语句会自动处理此问题，但也可以使用 close（）方法手动关闭文件。

异常处理：管理文件错误

文件操作容易出错，例如尝试打开不存在的文件或写入只读文件。Python 的异常处理机制允许我们优雅地管理这些错误。

示例：处理 File Not Found 错误

try:
    with open('non_existent_file.txt', 'r') as file:
        content = file.read()
except FileNotFoundError:
    print("The file does not exist.")

在此示例中，尝试打开一个不存在的文件，捕获 FileNotFoundError 并提供用户友好的消息。

高级文件处理技术

使用二进制文件

二进制文件以不适合人类阅读的格式存储数据。Python 可以使用 'b' 模式处理二进制文件。

示例：写入和读取二进制数据

# Writing binary data
with open('binary_file.bin', 'wb') as file:
    file.write(b'\xDE\xAD\xBE\xEF')

# Reading binary data
with open('binary_file.bin', 'rb') as file:
    data = file.read()
    print(data)

在此示例中，将十六进制数据写入二进制文件，然后将其读回，展示了 Python 处理非文本数据的能力。

处理大文件：高效处理

处理大型文件需要有效管理内存和处理时间的策略。以块形式读取文件或使用内存映射文件等技术可以显著提高性能。

示例：以块的形式读取大型文件

def read_large_file(file_path, chunk_size=1024):
    with open(file_path, 'r') as file:
        while chunk := file.read(chunk_size):
            process_chunk(chunk)  # Replace with actual processing logic

read_large_file('large_file.txt')

在这里，文件以 1024 字节的块读取，从而减少了内存使用，并允许我们处理大文件而不会使系统不堪重负。

上下文管理器：简化文件操作

Python 的上下文管理器与 with 语句一起使用，通过自动管理资源分配和取消分配来简化文件处理。

示例：用于文件处理的自定义上下文管理器

from contextlib import contextmanager

@contextmanager
def open_file(file_name, mode):
    file = open(file_name, mode)
    try:
        yield file
    finally:
        file.close()

with open_file('example.txt', 'r') as file:
    content = file.read()
    print(content)

在此示例中，我们定义了一个自定义上下文管理器，它模拟 Python 的内置文件处理，从而提供对文件操作的更多控制。