如何使用typing.io在Python中读取和写入CSV文件
发布时间:2024-01-19 02:20:15
在Python中读取和写入CSV文件可以使用csv模块,该模块提供了方便的方法来处理CSV格式的文件。下面是一个使用csv模块读取和写入CSV文件的例子。
首先,需要导入csv模块:
import csv
### 读取CSV文件
要读取CSV文件,可以使用csv.reader函数。这个函数需要接收一个打开的CSV文件对象作为参数,并返回一个reader对象,可以迭代读取CSV文件的每一行。以下是一个读取CSV文件的例子:
with open('data.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
# 操作每一行的数据
print(row)
上述代码假设有一个名为data.csv的CSV文件,reader对象可以迭代读取文件的每一行,并将每一行解析为一个列表(每个元素对应文件中的一个字段)。
### 写入CSV文件
要写入CSV文件,可以使用csv.writer函数。这个函数需要接收一个打开的CSV文件对象和一个可选的dialect参数,返回一个writer对象,可以使用writerow方法写入一行CSV数据。以下是一个写入CSV文件的例子:
data = [
['Name', 'Age', 'Gender'],
['John', '25', 'Male'],
['Jane', '30', 'Female']
]
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
上述代码将一个名为data.csv的CSV文件创建到当前目录中,并将列表data的每个子列表写入作为每一行的数据。
### 自定义分隔符
默认情况下,csv.reader和csv.writer使用逗号作为字段的分隔符。如果CSV文件中的字段分隔符不是逗号,可以使用csv.reader(file, delimiter='分隔符')和csv.writer(file, delimiter='分隔符')来自定义分隔符。
例如,如果CSV文件使用制表符作为字段的分隔符,可以这样读取和写入CSV文件:
with open('data.csv', 'r') as file:
reader = csv.reader(file, delimiter='\t')
for row in reader:
# 操作每一行的数据
print(row)
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file, delimiter='\t')
writer.writerows(data)
上述代码中,delimiter='\t'将制表符作为字段的分隔符。
以上就是使用csv模块读取和写入CSV文件的简单例子。希望对你有所帮助!
