import csv
打开CSV文件进行读取
with open('example.csv', mode='r') as file:
# 创建一个CSV阅读器对象
csv_reader = csv.reader(file)
# 遍历每一行
for row in csv_reader:
print(row)
写入CSV文件
import csv
数据列表
data = [['Name', 'Age', 'City'],
['Alice', 25, 'New York'],
['Bob', 30, 'San Francisco'],
['Charlie', 35, 'Los Angeles']]
打开CSV文件进行写入
with open('output.csv', mode='w', newline='') as file:
# 创建一个CSV写入器对象
csv_writer = csv.writer(file)
# 写入数据
for row in data:
csv_writer.writerow(row)
使用pandas读写CSV文件
pandas库提供了更高级的功能,可以更方便地处理复杂的数据操作。
安装pandas库
如果你还没有安装pandas,可以使用以下命令进行安装:
pip3 install pandas
读取CSV文件
import pandas as pd
读取CSV文件
df = pd.read_csv('example.csv')
显示前五行数据
print(df.head())