【Python】Python基础教程：CSV文件的写入与读取

CSV（Comma-Separated Values，逗号分隔值）文件是一种常见的数据存储格式，广泛用于数据交换和数据处理。Python 内置的 csv 模块提供了强大的功能来处理 CSV 文件。本文将详细介绍如何使用 Python 读取和写入 CSV 文件，并提供一些示例代码，帮助你在实际工作中充分利用这一工具。

🧑 博主简介：现任阿里巴巴嵌入式技术专家，15年工作经验，深耕嵌入式+人工智能领域，精通嵌入式领域开发、技术管理、简历招聘面试。CSDN优质创作者，提供产品测评、学习辅导、简历面试辅导、毕设辅导、项目开发、C/C++/Java/Python/Linux/AI等方面的服务，如有需要请站内私信或者联系任意文章底部的的VX名片（ID：gylzbk）

💬 博主粉丝群介绍：① 群内高中生、本科生、研究生、博士生遍布，可互相学习，交流困惑。② 热榜top10的常客也在群里，也有数不清的万粉大佬，可以交流写作技巧，上榜经验，涨粉秘籍。③ 群内也有职场精英，大厂大佬，可交流技术、面试、找工作的经验。④ 进群免费赠送写作秘籍一份，助你由写作小白晋升为创作大佬。⑤ 进群赠送CSDN评论防封脚本，送真活跃粉丝，助你提升文章热度。有兴趣的加文末联系方式，备注自己的CSDN昵称，拉你进群，互相学习共同进步。

1️⃣ CSV文件概述

CSV 文件是一种纯文本文件，每行代表一条记录，记录中的字段使用逗号分隔。例如：

name,age,city
Alice,30,New York
Bob,25,Los Angeles
Charlie,35,Chicago

2️⃣ 读取CSV文件

2.1 使用csv.reader读取CSV文件

csv.reader 可以将 CSV 文件逐行读取，并将每行的数据解析成一个列表。以下是一个简单的示例：

import csv

# 打开CSV文件
with open('example.csv', 'r', newline='') as csvfile:
    csvreader = csv.reader(csvfile)
    
    # 读取表头
    header = next(csvreader)
    print(f'Header: {header}')
    
    # 读取数据行
    for row in csvreader:
        print(row)

假设 example.csv 文件内容如下：

name,age,city
Alice,30,New York
Bob,25,Los Angeles
Charlie,35,Chicago

运行上述代码将输出：

Header: ['name', 'age', 'city']
['Alice', '30', 'New York']
['Bob', '25', 'Los Angeles']
['Charlie', '35', 'Chicago']

2.2 使用csv.DictReader读取CSV文件

csv.DictReader 将 CSV 文件的每一行解析成一个字典，键为表头，值为对应的字段值。以下是一个示例：

import csv

# 打开CSV文件
with open('example.csv', 'r', newline='') as csvfile:
    csvreader = csv.DictReader(csvfile)
    
    # 读取数据行
    for row in csvreader:
        print(row)

运行上述代码将输出：

{'name': 'Alice', 'age': '30', 'city': 'New York'}
{'name': 'Bob', 'age': '25', 'city': 'Los Angeles'}
{'name': 'Charlie', 'age': '35', 'city': 'Chicago'}

3️⃣ 写入CSV文件

3.1 使用csv.writer写入CSV文件

csv.writer 可以将数据写入 CSV 文件，数据以列表形式提供。以下是一个简单的示例：

import csv

# 要写入的数据
data = [
    ['name', 'age', 'city'],
    ['Alice', '30', 'New York'],
    ['Bob', '25', 'Los Angeles'],
    ['Charlie', '35', 'Chicago']
]

# 打开CSV文件
with open('output.csv', 'w', newline='') as csvfile:
    csvwriter = csv.writer(csvfile)
    
    # 写入数据
    csvwriter.writerows(data)

3.2 使用csv.DictWriter写入CSV文件

csv.DictWriter 可以将字典数据写入 CSV 文件，键为表头，值为对应的字段值。以下是一个示例：

import csv

# 要写入的数据
data = [
    {'name': 'Alice', 'age': 30, 'city': 'New York'},
    {'name': 'Bob', 'age': 25, 'city': 'Los Angeles'},
    {'name': 'Charlie', 'age': 35, 'city': 'Chicago'}
]

# 打开CSV文件
with open('output.csv', 'w', newline='') as csvfile:
    fieldnames = ['name', 'age', 'city']
    csvwriter = csv.DictWriter(csvfile, fieldnames=fieldnames)
    
    # 写入表头
    csvwriter.writeheader()
    
    # 写入数据
    csvwriter.writerows(data)

4️⃣ 高级用法

4.1 处理不同的分隔符

默认情况下，CSV 文件使用逗号作为分隔符。csv 模块允许你指定不同的分隔符，例如制表符（\t）：

4.1.1 读取使用制表符分隔的文件

import csv

with open('example.tsv', 'r', newline='') as tsvfile:
    tsvreader = csv.reader(tsvfile, delimiter='\t')
    for row in tsvreader:
        print(row)

4.1.2 写入使用制表符分隔的文件

import csv

data = [
    ['name', 'age', 'city'],
    ['Alice', '30', 'New York'],
    ['Bob', '25', 'Los Angeles'],
    ['Charlie', '35', 'Chicago']
]

with open('output.tsv', 'w', newline='') as tsvfile:
    tsvwriter = csv.writer(tsvfile, delimiter='\t')
    tsvwriter.writerows(data)

4.2 处理带有引号的字段

有时，CSV 文件中的字段可能包含逗号或其他特殊字符，这时我们可以使用引号将这些字段括起来。csv 模块可以自动处理这些情况。

4.2.1 读取带有引号的字段

import csv

with open('quoted_example.csv', 'r', newline='') as csvfile:
    csvreader = csv.reader(csvfile, quotechar='"', quoting=csv.QUOTE_ALL)
    for row in csvreader:
        print(row)

假设 quoted_example.csv 文件内容如下：

"name","age","city"
"Alice, A.","30","New York"
"Bob","25","Los Angeles"
"Charlie","35","Chicago"

4.2.2 写入带有引号的字段

import csv

data = [
    ['name', 'age', 'city'],
    ['Alice, A.', '30', 'New York'],
    ['Bob', '25', 'Los Angeles'],
    ['Charlie', '35', 'Chicago']
]

with open('quoted_output.csv', 'w', newline='') as csvfile:
    csvwriter = csv.writer(csvfile, quotechar='"', quoting=csv.QUOTE_ALL)
    csvwriter.writerows(data)

5️⃣ 总结

在这篇博客中，我们详细介绍了如何使用 Python 读取和写入 CSV 文件。具体内容包括：

• 使用 csv.reader 和 csv.DictReader 读取 CSV 文件。
• 使用 csv.writer 和 csv.DictWriter 写入 CSV 文件。
• 处理不同分隔符的 CSV 文件。
• 处理带有引号的字段。

通过这些示例，可以轻松地在 Python 中处理 CSV 文件，进行数据交换和数据处理。