【Python基础】一文搞懂：Python 中 csv 文件的写入与读取

在现代编程中，CSV（Comma-Separated Values）文件格式因其简单和通用性广泛应用于数据存储与交换。Python 提供了强大的内置库 csv 来处理这类文件。本文将详细介绍如何在 Python 中进行 CSV 文件的读取和写入操作。

一、CSV 文件的基本概念

CSV 文件是一种以逗号分隔的文本文件，每一行代表一条记录，行内的每个字段用逗号分开。CSV 文件通常用于存储表格数据，例如电子表格或数据库的导出文件。

二、写入 CSV 文件

在 Python 中写入 CSV 文件通常使用 csv 模块中的 writer 对象。以下是一个简单的例子，演示如何将数据写入 CSV 文件。

import csv

# 要写入的数据
data = [
    ["姓名", "年龄", "城市"],
    ["张三", 25, "北京"],
    ["李四", 30, "上海"],
    ["王五", 28, "广州"],
]

# 打开文件，准备写入
with open('example.csv', mode='w', newline='', encoding='utf-8') as file:
    writer = csv.writer(file)

    # 写入多行数据
    writer.writerows(data)

print("CSV 文件写入成功！")

在这个示例中，我们创建了一个包含多个记录的列表 data，然后使用 csv.writer 创建一个写入器对象。通过 writer.writerows 方法可以将数据一次性写入 CSV 文件中。

三、读取 CSV 文件

读取 CSV 文件同样使用 csv 模块，但我们需要用到 reader 对象。下面是一个读取 CSV 文件的例子：

import csv

# 打开文件，准备读取
with open('example.csv', mode='r', encoding='utf-8') as file:
    reader = csv.reader(file)

    # 逐行读取数据
    for row in reader:
        print(row)

在这个示例中，我们打开之前创建的 example.csv 文件，并创建了一个 reader 对象。通过迭代 reader，我们可以逐行读取 CSV 文件中的数据，row 将是一个列表，包含当前行的所有字段。

四、处理 CSV 字段

有时我们需要处理 CSV 文件中的特殊情况，比如字段中包含逗号或引号等。这时，我们可以调整 csv.writer 和 csv.reader 的参数。例如，在写入字段时，可以指定分隔符、引号字符等。

import csv

data = [
    ["姓名", "年龄", "城市"],
    ['张 "三"', 25, "北京"],
    ["李四", 30, "上海,浦东"],
]

with open('example_handling.csv', mode='w', newline='', encoding='utf-8') as file:
    writer = csv.writer(file, quoting=csv.QUOTE_ALL)
    writer.writerows(data)

print("处理后的 CSV 文件写入成功！")

在这个例子中，我们使用 quoting=csv.QUOTE_ALL 参数来确保所有字段都用引号包裹，以避免字段中的特殊字符导致的问题。

五、总结

CSV 文件以其易读性和简单性被广泛应用于数据交换。Python 的 csv 模块提供了便利的读取和写入功能，使得程序员能够轻松处理 CSV 文件。通过设置不同的参数，还能应对各种特殊情况。掌握 CSV 文件的处理方式，将极大提升数据处理的效率。

这篇文章简单介绍了 Python 中 CSV 文件的基本操作，希望对你有所帮助！