Python如何读取和写入CSV文件的函数?
发布时间:2023-08-22 07:08:31
在Python中,可以使用csv模块来读取和写入CSV文件。
要读取CSV文件,我们可以使用csv.reader函数。首先,需要打开CSV文件,然后使用csv.reader函数来创建一个reader对象。可以使用reader对象的readrows()方法来逐行读取CSV文件中的数据。以下是一个例子:
import csv
# 打开CSV文件
with open('file.csv', 'r') as file:
# 创建reader对象
reader = csv.reader(file)
# 读取数据
for row in reader:
print(row)
要写入CSV文件,我们可以使用csv.writer函数。首先,需要打开CSV文件,然后使用csv.writer函数来创建一个writer对象。可以使用writer对象的writerow()方法来逐行写入数据。以下是一个例子:
import csv
# 要写入的数据
data = [['Name', 'Age'], ['John Doe', 30], ['Jane Smith', 25]]
# 打开CSV文件
with open('file.csv', 'w', newline='') as file:
# 创建writer对象
writer = csv.writer(file)
# 写入数据
writer.writerows(data)
值得注意的是,在写入CSV文件时,需要使用newline=''参数来避免插入空行。
可以使用csv模块的其他函数和方法来处理CSV文件,如csv.DictReader和csv.DictWriter来处理包含标题行的CSV文件,和csv.writer的writerow()和writerows()方法来逐行写入数据。
总结起来,要读取和写入CSV文件,可以使用csv.reader和csv.writer函数,它们提供了方便的方法来处理CSV文件中的数据。
