一、writerow方法基础
writer.writerow
是 Python 内置 csv 模块中的一个用于写入 csv 文件的方法。它的作用是将一个列表或元组中的数据写入到文件中,并自动将数据以 csv 格式进行存储。
import csv
# 定义csv文件名称
file_name = 'example.csv'
# 定义csv文件打开方式
mode = 'w'
# 创建csv写入对象
with open(file_name, mode, newline='', encoding='utf-8') as csv_file:
# 定义csv写入对象
csv_writer = csv.writer(csv_file)
# 将数据写入到文件中
csv_writer.writerow(['Name', 'Age', 'Score'])
csv_writer.writerow(['张三', 20, 98])
csv_writer.writerow(['李四', 22, 95])
以上代码实现了将一个具有三列数据(姓名、年龄、成绩)的 csv 文件生成,其中包括两行数据。
二、writerow方法中的参数和用法
writer.writerow
方法可以传入多个参数,可以是元组、列表或包含多个元组或列表的可迭代对象等。以下给出 writer.writerow
方法中的几种常见参数类型并简要介绍它们的使用。
1. 以列表为参数
使用列表作为参数,该方法将列表中的元素作为 csv 文件中的一行数据写入到文件中。这里需要注意的是,传递的列表中的元素必须是可迭代的类型(List、Tuple 等)。示例代码如下:
import csv
# 定义csv文件名称
file_name = 'example1.csv'
# 定义csv文件打开方式
mode = 'w'
# 创建csv写入对象
with open(file_name, mode, newline='', encoding='utf-8') as csv_file:
# 定义csv写入对象
csv_writer = csv.writer(csv_file)
# 将数据写入到文件中
row1 = ['Name', 'Age', 'Score']
row2 = ['张三', 20, 98]
row3 = ['李四', 22, 95]
csv_writer.writerow(row1)
csv_writer.writerow(row2)
csv_writer.writerow(row3)
2. 以元组为参数
与使用列表作为参数一样,使用元组作为参数,将元组中的元素作为 csv 行写入到文件中。示例代码如下:
import csv
# 定义csv文件名称
file_name = 'example2.csv'
# 定义csv文件打开方式
mode = 'w'
# 创建csv写入对象
with open(file_name, mode, newline='', encoding='utf-8') as csv_file:
# 定义csv写入对象
csv_writer = csv.writer(csv_file)
# 将数据写入到文件中
row1 = ('Name', 'Age', 'Score')
row2 = ('张三', 20, 98)
row3 = ('李四', 22, 95)
csv_writer.writerow(row1)
csv_writer.writerow(row2)
csv_writer.writerow(row3)
3. 使用可迭代对象
可迭代对象可以是包含多个元组或列表的容器,如列表、元组、生成器等。该方法将可迭代对象中的每个元素作为一个 csv 行写入到文件中。示例代码如下:
import csv
# 定义csv文件名称
file_name = 'example3.csv'
# 定义csv文件打开方式
mode = 'w'
# 创建csv写入对象
with open(file_name, mode, newline='', encoding='utf-8') as csv_file:
# 定义csv写入对象
csv_writer = csv.writer(csv_file)
# 将数据写入到文件中
data = [
['Name', 'Age', 'Score'],
['张三', 20, 98],
['李四', 22, 95]
]
csv_writer.writerows(data)
三、writerow方法的注意事项
writer.writerow
方法虽然方便了我们对 csv 文件的读写,但在使用时也需要注意一些细节问题。
1. 中文字符编码问题
在 Python 中,csv 文件的编码格式默认为 'utf-8'
,但在有些情况下,中文字符可能会出现乱码现象。解决方法是在文件打开时指定编码格式,如:
import csv
# 定义csv文件名称
file_name = 'example4.csv'
# 定义csv文件打开方式
mode = 'w'
# 创建csv写入对象
with open(file_name, mode, newline='', encoding='gbk') as csv_file:
# 定义csv写入对象
csv_writer = csv.writer(csv_file)
# 将数据写入到文件中
row1 = ['姓名', '年龄', '成绩']
row2 = ['张三', 20, 98]
row3 = ['李四', 22, 95]
csv_writer.writerow(row1)
csv_writer.writerow(row2)
csv_writer.writerow(row3)
2. 避免空行出现
当使用 writerow
方法写入多行数据时,有可能出现空行的情况。这是因为 writerow
方法默认在每次写入数据后会自动添加一个空行。避免空行出现的方法是在文件打开时使用 newline=''
参数。示例代码如下:
import csv
# 定义csv文件名称
file_name = 'example5.csv'
# 定义csv文件打开方式
mode = 'w'
# 创建csv写入对象
with open(file_name, mode, newline='', encoding='utf-8') as csv_file:
# 定义csv写入对象
csv_writer = csv.writer(csv_file)
# 将数据写入到文件中
row1 = ['Name', 'Age', 'Score']
row2 = ['张三', 20, 98]
row3 = ['李四', 22, 95]
csv_writer.writerow(row1)
csv_writer.writerow(row2)
csv_writer.writerow(row3)
3. 尽量避免出现特殊字符
writer.writerow
方法是将数据以 csv 格式写入到文件中,其中 csv 格式中使用逗号分隔数据项。因此,如果被写入的数据中存在逗号、引号、换行符等特殊字符,将会影响到数据的正确读取。
四、总结
本文主要介绍了 Python 内置 csv 模块中的 writer.writerow
方法的使用方法和注意事项。该方法非常方便我们对 csv 文件进行读写操作,在实际应用中也非常常见。希望本文能够为大家提供到帮助。