在Python中导入CSV数据主要有以下两种方式,根据需求选择合适的方法:
一、使用内置 csv
模块(基础操作)
适用于简单场景,无需额外安装库。
读取CSV文件:
import csv
with open('data.csv', newline='', encoding='utf-8') as csvfile:
reader = csv.reader(csvfile)
for row in reader:
print(row)
newline=''
避免Windows系统出现多余空行。
写入CSV文件:
import csv
data = [
['Name', 'Age', 'City'],
['Alice', '30', 'New York'],
['Bob', '25', 'San Francisco']
]
with open('output.csv', 'w', newline='', encoding='utf-8') as csvfile:
writer = csv.writer(csvfile)
writer.writerows(data)
二、使用 pandas
库(高效处理)
适合复杂数据分析场景,支持多种数据格式转换。
读取CSV文件:
import pandas as pd
df = pd.read_csv('data.csv')
print(df.head())
- 支持指定列名、跳过行、分块读取等高级功能。
写入CSV文件:
df.to_csv('output.csv', index=False)
index=False
可避免导出DataFrame索引列。
总结建议
-
简单读写 :优先使用
csv
模块,轻量且无需依赖外部库。 -
数据分析 :推荐使用
pandas
,功能强大且兼容性高。