Python常用内置模块-csv模块

Mark_Coder2024-07-20 19:10

介绍

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'})