博客

csv是Python内置的处理CSV文件的模块，提供了一系列方便易用的读写CSV文件的函数和类。其中，writer函数是csv模块中用于写入CSV文件的函数。

安装和导入CSV模块

csv模块属于Python内置模块，不需要额外安装，可以直接在代码中导入使用：

import csv

writer函数的基本语法

writer函数是csv模块中提供的写入CSV文件的函数，语法格式如下：

csv.writer(csvfile, dialect='excel', **fmtparams)

其中，参数解释如下：

• csvfile：要写入数据的文件对象，可以是打开的文件或者其他可支持写入的文件对象；
• dialect：CSV文件的格式，不同的格式对应不同的CSV文件类型，常用Dialect格式包括excel, excel-tab, unix等；
• fmtparams：其他参数，可以用于设置写入CSV文件的格式。

writer函数的使用方法

使用csv.writer()函数进行csv文件的写入时，需要完成以下操作：

1. 打开或创建要写入的CSV文件；
2. 创建writer对象；
3. 通过writerow()函数将数据写入CSV文件；
4. 关闭CSV文件。

下面是使用writer函数进行csv文件写入的示例代码：

import csv

# 创建并打开CSV文件，如果不存在则创建
with open('test.csv', 'w', newline='') as csvfile:
    # 创建writer对象，指定CSV文件的格式为excel
    writer = csv.writer(csvfile, dialect='excel')

    # 写入数据
    writer.writerow(['name', 'age', 'score'])    # 写入表头
    writer.writerow(['Tom', '18', '98'])    # 写入一行数据
    writer.writerow(['Lucy', '20', '85'])

    # 关闭CSV文件
    csvfile.close()

在上述示例代码中，首先通过open()函数创建CSV文件对象，然后通过csv.writer()函数创建writer对象用于写入CSV文件。在写入数据时，使用writer.writerow()函数将数据写入CSV文件，每一行数据用一个列表表示。在写完所有数据后，关闭CSV文件。

另外，还可以使用writerows()函数将多行数据一次性写入CSV文件：

import csv

# 创建并打开CSV文件，如果不存在则创建
with open('test.csv', 'w', newline='') as csvfile:
    # 创建writer对象，指定CSV文件的格式为excel
    writer = csv.writer(csvfile, dialect='excel')

    # 写入数据
    writer.writerow(['name', 'age', 'score'])    # 写入表头
    writerows = []
    writerows.append(['Tom', '18', '98'])    # 添加一行数据
    writerows.append(['Lucy', '20', '85'])
    writer.writerows(writerows)    # 将writerows中的所有数据写入CSV文件

    # 关闭CSV文件
    csvfile.close()

在上述示例代码中，首先通过open()函数创建CSV文件对象，然后通过csv.writer()函数创建writer对象用于写入CSV文件。在写入数据时，首先使用writer.writerow()函数写入表头，然后通过列表writerows存储所有数据，并通过writer.writerows()函数将writerows中的所有行数据一次性写入CSV文件。正如前面提到的，每一行数据用一个列表表示。

总结

使用csv模块的writer函数可以很方便地将数据写入CSV文件中，可以通过writerow()和writerows()函数实现单行和多行数据的写入，同时还可以通过参数配置设置CSV文件的格式等。在使用writer函数时，需要学会打开和关闭CSV文件，以免出现文件写入错误等问题。