Python csv写入详解|极客教程

Python csv写入详解

在Python中，通过使用csv模块，我们可以很方便地操作CSV（Comma Separated Values）文件格式，即逗号分隔值的文件。CSV文件是一种常见的数据存储格式，常常用于数据导入导出、数据交换等场景。在本文中，我们将详细介绍如何使用Python的csv模块进行CSV文件的写入操作。

1. CSV文件结构

在CSV文件中，每行代表一条记录，每条记录的数据项使用逗号进行分隔。以下是一个示例的CSV文件结构：

Name, Age, Gender
Alice, 25, Female
Bob, 30, Male
Charlie, 35, Male

在这个示例中，每行代表一个人的信息，包括姓名、年龄和性别，其中逗号用于分隔不同的数据项。

2. 使用csv模块写入CSV文件

在Python中，可以通过csv模块来创建和写入CSV文件。下面是一个简单的示例，演示了如何使用csv.writer类来写入一个CSV文件：

import csv

data = [
    ['Name', 'Age', 'Gender'],
    ['Alice', 25, 'Female'],
    ['Bob', 30, 'Male'],
    ['Charlie', 35, 'Male']
]

with open('example.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

print("CSV file has been written successfully!")

在这个示例中，我们首先构造了一个包含数据的列表data，然后通过csv.writer类创建了一个writer对象。接着，我们使用writer.writerows(data)方法将数据写入到名为example.csv的CSV文件中。最后通过打印输出操作的成功完成。

3. CSV写入工作原理

在上面的示例中，我们简单介绍了如何使用csv模块写入CSV文件。要更深入地理解写入操作的工作原理，我们来解析一下上面的代码：

with open('example.csv', 'w', newline='') as file:：打开一个名为example.csv的文件，并指定使用写入模式（’w’）。newline=''参数是为了确保在写入CSV文件时，不会多出一个空行。
writer = csv.writer(file)：创建一个writer对象，用于将数据写入到CSV文件中。
writer.writerows(data)：通过writerows方法将数据data写入到CSV文件中。

4. CSV写入进阶操作

除了上面的基本写入操作外，csv模块还提供了一些更高级的写入功能，比如：

4.1. 指定分隔符

除了默认的逗号分隔符，我们也可以指定其他分隔符。例如，将数据写入以制表符为分隔符的CSV文件：

import csv

data = [
    ['Name', 'Age', 'Gender'],
    ['Alice', 25, 'Female'],
    ['Bob', 30, 'Male'],
    ['Charlie', 35, 'Male']
]

with open('example.tsv', 'w', newline='') as file:
    writer = csv.writer(file, delimiter='\t')
    writer.writerows(data)

print("TSV file has been written successfully!")

在这个示例中，我们通过delimiter='\t'参数指定了制表符（\t）为分隔符，来创建一个以制表符分隔的TSV文件。

4.2. 写入字典数据

除了列表数据外，我们也可以写入字典数据到CSV文件。下面是一个示例，演示了如何写入字典数据：

import csv

data = [
    {'Name': 'Alice', 'Age': 25, 'Gender': 'Female'},
    {'Name': 'Bob', 'Age': 30, 'Gender': 'Male'},
    {'Name': 'Charlie', 'Age': 35, 'Gender': 'Male'}
]

fieldnames = ['Name', 'Age', 'Gender']

with open('example_dict.csv', 'w', newline='') as file:
    writer = csv.DictWriter(file, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerows(data)

print("CSV file with dictionary data has been written successfully!")

在这个示例中，我们通过csv.DictWriter类将字典数据写入到CSV文件中，并通过writeheader()方法写入字段名作为文件的第一行。