如何在Python中读写CSV文件?
发布时间:2023-06-24 14:05:48
CSV(Comma-Separated Values)是一种常用的文件格式,它将数据以逗号的方式分隔开,一般用于存储表格数据。Python中有多种方式可以读取和写入CSV文件,本文将介绍其中的两种方式:
1. 使用csv模块
CSV模块在Python的标准库中。它提供了一组方法和类,可以很方便地读取和写入CSV文件。下面是使用CSV模块读取CSV文件的示例代码:
import csv
with open('data.csv', 'r') as f:
reader = csv.reader(f)
for row in reader:
print(row)
解析代码:
- csv.reader()方法用于读取CSV文件的数据,并返回一个可迭代的reader对象。
- for循环用于遍历可迭代的reader对象。
- 打印每一行的数据。
下面是使用CSV模块写入CSV文件的示例代码:
import csv
data = [
['Name', 'Age', 'Gender'],
['John', '25', 'Male'],
['Jane', '30', 'Female'],
['Bob', '35', 'Male']
]
with open('data.csv', 'w', newline='') as f:
writer = csv.writer(f)
for row in data:
writer.writerow(row)
解析代码:
- 使用csv.writer()方法创建一个CSV写入对象。
- for循环用于遍历data列表中的每一行数据。
- writer.writerow()方法用来写入一行数据到CSV文件中。
2. 使用pandas模块
pandas是Python的一种数据分析库,提供了更好的API来读取和写入各种格式的文件,包括CSV文件。下面是使用pandas模块读取CSV文件的示例代码:
import pandas as pd
df = pd.read_csv('data.csv')
print(df)
解析代码:
- pd.read_csv()方法用于读取CSV文件,并返回一个DataFrame对象。
- DataFrame对象是pandas中的一种数据结构,类似于Excel表格。
- 打印DataFrame对象。
下面是使用pandas模块写入CSV文件的示例代码:
import pandas as pd
data = {
'Name': ['John', 'Jane', 'Bob'],
'Age': [25, 30, 35],
'Gender': ['Male', 'Female', 'Male']
}
df = pd.DataFrame(data)
df.to_csv('data.csv', index=False)
解析代码:
- pd.DataFrame()方法用于从数据集构建一个DataFrame对象。
- DataFrame对象是pandas中的一种数据结构,类似于Excel表格。
- df.to_csv()方法用于将DataFrame对象写入CSV文件中。
