Python-csv库进行数据保存和读写

在 Python 中使用 CSV 文件非常简单，Python 提供了内置的 csv 模块来处理 CSV 文件。你可以使用 csv 模块来读取、写入和操作 CSV 文件中的数据。

基础使用

读取 CSV 文件

python
import csv

# 打开 CSV 文件进行读取
with open('file.csv', mode='r') as file:
    reader = csv.reader(file)  # 创建 CSV 读取器对象
    for row in reader:
        print(row)  # 逐行打印 CSV 文件中的数据

写入 CSV 文件

python
import csv

# 写入数据到 CSV 文件
data = [
    ['Name', 'Age', 'Gender'],
    ['Alice', 25, 'Female'],
    ['Bob', 28, 'Male'],
    ['Cathy', 22, 'Female']
]

with open('output.csv', mode='w', newline='') as file:
    writer = csv.writer(file)  # 创建 CSV 写入器对象
    writer.writerows(data)  # 将数据写入到 CSV 文件

在这两个示例中，首先需要导入 csv 模块。使用 with open() 打开 CSV 文件并指定文件模式（'r' 表示读取，'w' 表示写入）。然后使用 csv.reader() 或 csv.writer() 创建读取器或写入器对象。读取器可以逐行读取 CSV 文件的内容，写入器可以将数据写入到 CSV 文件中。

更多使用技巧

1. 处理不同格式的分隔符

有时 CSV 文件中的字段可能不是用逗号分隔的，可能会使用其他字符作为分隔符，比如制表符 \t。你可以在读取和写入时指定不同的分隔符。

读取不同分隔符的 CSV 文件：

# 使用制表符作为分隔符读取 CSV 文件
with open('file.tsv', mode='r') as file:
    reader = csv.reader(file, delimiter='\t')  # 指定分隔符为制表符
    for row in reader:
        print(row)

写入不同分隔符的 CSV 文件：

# 使用分号作为分隔符写入 CSV 文件
data = [
    ['Name', 'Age', 'Gender'],
    ['Alice', 25, 'Female'],
    ['Bob', 28, 'Male'],
    ['Cathy', 22, 'Female']
]

with open('output.csv', mode='w', newline='') as file:
    writer = csv.writer(file, delimiter=';')  # 指定分隔符为分号
    writer.writerows(data)

2. 处理包含引号的数据

有些 CSV 文件中的字段可能包含引号，这时在处理时可能会出现问题。你可以指定引号的处理方式，以便正确读取包含引号的字段。

# 处理包含引号的数据
with open('file.csv', mode='r') as file:
    reader = csv.reader(file, quoting=csv.QUOTE_MINIMAL)  # 指定引号处理方式
    for row in reader:
        print(row)

3. 处理文件编码

在处理 CSV 文件时，特别是处理非英文字符时，确保指定文件的正确编码方式，以免出现乱码问题。

# 指定文件编码方式
with open('file.csv', mode='r', encoding='utf-8') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

4. 考虑使用 DictReader 和 DictWriter

除了 csv.reader() 和 csv.writer() 外，csv 模块还提供了 csv.DictReader() 和 csv.DictWriter()，它们可以将每一行数据作为字典进行处理，使用列标题作为字典的键。

# 使用 DictReader 读取 CSV 文件
with open('file.csv', mode='r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row['Name'], row['Age'], row['Gender'])
python
# 使用 DictWriter 写入 CSV 文件
fieldnames = ['Name', 'Age', 'Gender']
data = [
    {'Name': 'Alice', 'Age': 25, 'Gender': 'Female'},
    {'Name': 'Bob', 'Age': 28, 'Gender': 'Male'},
    {'Name': 'Cathy', 'Age': 22, 'Gender': 'Female'}
]

with open('output.csv', mode='w', newline='') as file:
    writer = csv.DictWriter(file, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerows(data)

这些技巧可以帮助你更好地使用 csv 模块处理不同格式的 CSV 文件，同时避免一些常见的问题，如分隔符问题、引号处理和文件编码等。