首页 > 脚本专栏 > python > Python文件I/O

一文带你掌握Python中文件I/O的使用技巧

2023-12-28 09:11:42 作者：Sitin涛哥

Python提供了强大而灵活的文件I/O（输入/输出）工具,能够读取、写入和处理各种文件类型,本文将详细介绍Python文件I/O的使用技巧,需要的可以参考一下

Python提供了强大而灵活的文件I/O（输入/输出）工具，能够读取、写入和处理各种文件类型。本文将深入介绍Python文件I/O的技巧和示例代码，帮助大家更好地理解如何在Python中处理文件。

打开文件

在Python中，可以使用open()函数来打开文件，指定文件名和打开模式。常见的打开模式包括：

'r'：只读模式（默认），用于读取文件内容。

'w'：写入模式，用于创建新文件或覆盖已有文件。

'a'：追加模式，用于在文件末尾添加内容。

'b'：二进制模式，用于处理二进制文件（如图像、音频等）。

't'：文本模式（默认），用于处理文本文件。

以下是打开文件的示例代码：

# 打开一个文本文件以供读取
with open('example.txt', 'r') as file:
    data = file.read()
    print(data)

# 打开一个文本文件以供写入
with open('new_file.txt', 'w') as file:
    file.write('Hello, World!')

# 打开一个二进制文件以供读取
with open('image.jpg', 'rb') as file:
    image_data = file.read()

逐行读取文件

如果需要逐行读取文件内容，可以使用readline()方法或迭代文件对象。以下是示例代码：

# 使用readline()逐行读取文件
with open('example.txt', 'r') as file:
    while True:
        line = file.readline()
        if not line:
            break
        print(line.strip())  # 去除行末的换行符

# 使用迭代文件对象逐行读取
with open('example.txt', 'r') as file:
    for line in file:
        print(line.strip())

写入文件

要将数据写入文件，可以使用write()方法。如果想写入多行内容，可以使用writelines()方法。以下是示例代码：

# 写入文本到文件
with open('new_file.txt', 'w') as file:
    file.write('Line 1\n')
    file.write('Line 2\n')

# 写入多行文本到文件
lines = ['Line 1\n', 'Line 2\n', 'Line 3\n']
with open('new_file.txt', 'w') as file:
    file.writelines(lines)

处理CSV文件

CSV文件常用于存储表格数据。Python提供了csv模块来处理CSV文件。以下是一个读取CSV文件的示例：

import csv

with open('data.csv', 'r') as file:
    csv_reader = csv.reader(file)
    for row in csv_reader:
        print(row)

还可以使用csv模块来写入CSV文件。

使用上下文管理器

使用with语句和上下文管理器可以确保在文件使用完毕后自动关闭文件，无需手动调用close()方法。

# 使用上下文管理器来读取文件
with open('example.txt', 'r') as file:
    data = file.read()
    print(data)  # 在此块中，文件已自动关闭

# 使用上下文管理器来写入文件
with open('new_file.txt', 'w') as file:
    file.write('Hello, World!')  # 在此块中，文件已自动关闭

异常处理

在文件I/O操作中，异常处理是确保程序稳定性的重要部分。文件可能会出现各种问题，如文件不存在、权限问题、文件格式错误等。适当的异常处理可以避免程序因异常而崩溃，同时提供有用的错误信息。

以下是一个处理文件不存在异常的示例：

try:
    with open('non_existent_file.txt', 'r') as file:
        data = file.read()
        print(data)
except FileNotFoundError:
    print("文件不存在")

在这个示例中，使用try和except语句来捕获FileNotFoundError异常。如果文件不存在，程序将打印"文件不存在"，而不会中断执行。

除了FileNotFoundError，还有其他可能的异常，如PermissionError（权限问题）和IOError（一般I/O错误）。根据具体情况，可以选择处理不同类型的异常。

移动文件指针

在读取文件时，文件指针表示当前读取的位置。可以使用seek()方法来移动文件指针到文件中的特定位置。

例如，要从文件的开头开始读取，可以将文件指针移动到偏移量为0的位置：

with open('example.txt', 'r') as file:
    file.seek(0)  # 将文件指针移动到文件开头
    data = file.read()
    print(data)

使用os模块操作文件系统

os模块提供了许多与文件系统交互的功能，包括创建文件夹、删除文件、重命名文件等。

以下是一些示例：

import os

# 创建文件夹
os.mkdir('my_folder')

# 删除文件
os.remove('file_to_delete.txt')

# 重命名文件
os.rename('old_file.txt', 'new_file.txt')

# 获取文件列表
file_list = os.listdir('.')

使用shutil模块复制和移动文件

shutil模块提供了复制、移动和删除文件的功能。