读写CSV文件在Python中是一个常见的任务,可以使用内置的csv
模块或第三方库如pandas
来完成。以下是使用这两种方法的示例:
使用csv
模块
读取CSV文件
import csvwith open('example.csv', newline='') as csvfile:reader = csv.reader(csvfile)for row in reader:print(row)
写入CSV文件
import csvwith open('example.csv', 'w', newline='') as csvfile:writer = csv.writer(csvfile)writer.writerow(['Name', 'Age', 'City'])writer.writerow(['Alice', 30, 'New York'])writer.writerow(['Bob', 25, 'Los Angeles'])
使用pandas
库
读取CSV文件
import pandas as pddf = pd.read_csv('example.csv')
print(df)
写入CSV文件
import pandas as pddata = {'Name': ['Alice', 'Bob'],'Age': [30, 25],'City': ['New York', 'Los Angeles']
}
df = pd.DataFrame(data)
df.to_csv('example.csv', index=False)
选择合适的方法
csv
模块适合处理简单的CSV读写操作,尤其是当你只需要逐行读取或写入数据时。pandas
库提供了更强大的数据处理功能,适合处理复杂的数据分析任务。