使用Python编程写入CSV文件

小蓝 • 2024-11-29 13:54 • 编程

CSV（Comma Separated Values）文件是一种以文本形式存储表格数据的文件格式，数据之间使用逗号分隔。在日常工作中，我们常常需要将表格数据以CSV格式进行保存和导出，因此，学习如何使用Python编程将数据写入CSV文件是一项必备技能。

一、CSV文件的基本结构

CSV文件的基本结构由行和列组成，每行表示一组数据，每列表示一种数据类型，不同列之间用逗号隔开。一个简单的CSV文件示例如下：

姓名,性别,年龄
张三,男,22
李四,女,21
王五,男,23

在Python中，我们通常使用csv模块来读写CSV文件。csv模块提供了一个CSV文件读写器和CSV文件写入器，使得我们可以很方便地处理CSV文件。

二、使用Python编写CSV文件写入器

1. 编写CSV文件头

在写入CSV文件之前，我们需要先使用csv模块创建一个CSV文件写入器。CSV文件写入器有两个主要方法：writerow()和writerows()。writerow()方法用于写入一行CSV数据，writerows()方法用于写入多行CSV数据。在写入CSV数据之前，我们需要先写入CSV文件头，即第一行数据，表示每一列的字段名。

import csv

with open('test.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['姓名', '性别', '年龄'])

2. 写入CSV数据

写入CSV文件头之后，我们就可以向CSV文件中写入数据了。写入CSV数据时，我们需要先将数据按照每一行的格式封装成一个列表，然后通过writerow()方法写入CSV文件中。

import csv

with open('test.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['姓名', '性别', '年龄'])
    writer.writerow(['张三', '男', '22'])
    writer.writerow(['李四', '女', '21'])
    writer.writerow(['王五', '男', '23'])

运行以上代码，就可以生成一个名为test.csv的CSV文件，其内容如下：

姓名,性别,年龄
张三,男,22
李四,女,21
王五,男,23

三、扩展应用

除了基本的CSV文件写入操作，我们还可以使用csv模块实现更加复杂的CSV文件读写操作。例如，我们可以使用csv.DictWriter()方法创建一个CSV字典写入器，基于字典的方式写入CSV文件。代码示例：

import csv

with open('test.csv', 'w', newline='') as csvfile:
    fieldnames = ['姓名', '性别', '年龄']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerow({'姓名': '张三', '性别': '男', '年龄': '22'})
    writer.writerow({'姓名': '李四', '性别': '女', '年龄': '21'})
    writer.writerow({'姓名': '王五', '性别': '男', '年龄': '23'})

以上代码将生成与之前示例相同的test.csv文件，但是使用更加高级的CSV字典写入方式实现数据写入。

四、总结

Python提供了csv模块，使得我们可以很方便地读写CSV文件。对于需要频繁处理CSV文件的工程师来说，掌握CSV文件读写的基本操作是必不可少的。希望本文能够帮助大家了解如何使用Python编程写入CSV文件。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/190150.html