博客

当我们需要将数据写入到CSV文件中时，Python提供了一个非常方便的csv库。

csv库提供了一个writer对象，它可以让我们很方便的将数据写入到CSV文件中。下面就是csv库的writer函数的完整攻略：

1. 导入csv库

import csv

2. 打开CSV文件

在实际操作中，我们需要先打开CSV文件，可以通过以下方法打开：

with open('example.csv', 'w', newline='') as csvfile:
   writer = csv.writer(csvfile)

上述代码中，“example.csv”就是我们要打开的CSV文件名。

此外，newline参数是用于控制行之间的空行的显示情况。设置为空字符串就能解决绝大部分写入文件换行多一个空行问题。

3. 写入数据

接下来，就可以通过writer对象向CSV文件中写入数据了。如下代码所示：

with open('example.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['姓名', '性别', '年龄']) # 写入表头
    writer.writerow(['张三', '男', '18']) # 写入数据

以上代码，在CSV文件中写入了表头和一条数据。

需要特别注意的是，数据的保存格式是以列表的形式保存的，写入一行数据的方式为writer.writerow([‘value1′,’value2’,…])。如果要写入多行数据，多次调用该方法即可。

4. 关闭文件

最后，要记得关闭文件。通常情况下使用with，可以省去这一步。

# with方法自带了关闭文件的功能，不用特别进行关闭

详细讲解python中csv库的writer函数的完整攻略就介绍到这里了。