Python常用内置模块-csv模块

介绍

Python中的csv模块是用于处理CSV（逗号分隔值）文件的标准库。CSV文件是一种常见的文件格式，用于存储和交换数据。一般也被叫做逗号分隔值文件，可以使用 Excel 软件或者文本文档打开 。其中数据字段用半角逗号间隔（也可以使用其它字符），使用 Excel 打开时，逗号会被转换为分隔符。csv 文件是以纯文本形式存储了表格数据，并且在兼容各个操作系统。

使用

1. 导入csv模块：

import csv

2. 读取CSV文件：

• 打开CSV文件并创建一个读取器(reader)对象：

with open('data.csv', 'r') as file:
    reader = csv.reader(file)

• 读取文件中的数据行：

for row in reader:
    # 对每一行进行处理，row是一个列表，包含该行中的所有字段
    # 例如，可以使用row[index]访问特定字段
    print(row)

3. 写入CSV文件：

• 打开CSV文件并创建一个写入器(writer)对象：

with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)

• 写入数据行：

writer.writerow(['Name', 'Age', 'City'])  # 写入表头
writer.writerow(['John', '25', 'New York'])  # 写入数据行

• 写入多行数据：

data = [['Name', 'Age', 'City'], ['John', '25', 'New York'], ['Jane', '30', 'Los Angeles']]
	writer.writerows(data)

4. 读取CSV文件为字典：

• 打开CSV文件并创建一个字典读取器(DictReader)对象：

with open('data.csv', 'r') as file:
	reader = csv.DictReader(file)

• 读取文件中的数据行，每行将作为一个字典返回：

for row in reader:
    # 对每一行进行处理，row是一个字典，键为字段名，值为对应的值
    # 例如，可以使用row['Name']访问特定字段的值
	print(row)

5. 写入字典到CSV文件：

• 定义CSV文件的字段名：

fieldnames = ['Name', 'Age', 'City']

• 打开CSV文件并创建一个字典写入器(DictWriter)对象：

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

• 写入表头：

writer.writeheader()

• 写入字典数据行：

writer.writerow({'Name': 'John', 'Age': '25', 'City': 'New York'})