writer.writerow使用详解

发布时间:2023-05-24

一、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 文件进行读写操作,在实际应用中也非常常见。希望本文能够为大家提供到帮助。