在Python中,CSV(Comma Separated Values)文件是一种常见的数据存储格式,我们可以使用内置的csv
模块来实现对CSV文件的读写操作。
1. 写入CSV文件
import csv
# 打开或创建一个csv文件
with open('example.csv', 'w', newline='') as csvfile:
# 创建csv writer对象
writer = csv.writer(csvfile)
# 写入表头
writer.writerow(['Name', 'Age', 'City'])
# 写入多行数据
writer.writerow(['John', '30', 'New York'])
writer.writerow(['Jane', '25', 'Los Angeles'])
2. 读取CSV文件
import csv
# 打开csv文件
with open('example.csv', 'r') as csvfile:
# 创建csv reader对象
reader = csv.reader(csvfile)
# 跳过表头
next(reader) # 如果有表头的话
# 遍历并打印每一行数据
for row in reader:
print(row)
另外,对于更复杂的情况,如指定分隔符、处理Excel格式的CSV等,可以使用csv.DictReader
和csv.DictWriter
以字典形式读写CSV,或者使用pandas
库进行更高级的操作。