python 写入csv文件

python 写入csv文件

CSV（Comma-Separated Values）是一种常见的文本文件格式，用于将数据以逗号分隔的形式存储。在Python中，我们可以使用csv模块来操作CSV文件。本文将详细介绍如何使用Python写入CSV文件。

1. 导入csv模块

首先，我们需要导入Python的csv模块来操作CSV文件。可以使用以下代码导入csv模块：

import csv

2. 创建CSV文件并写入数据

接下来，我们可以使用csv.writer()方法创建一个csv写入对象，并使用writerow()方法写入数据。以下是一个示例代码：

# 创建CSV文件并写入数据
with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['Name', 'Age', 'City'])
    writer.writerow(['Alice', 25, 'New York'])
    writer.writerow(['Bob', 30, 'San Francisco'])
    writer.writerow(['Charlie', 35, 'Chicago'])

在上面的示例中，我们首先打开一个名为data.csv的文件并指定为写入模式。然后，我们创建一个csv写入对象writer，并使用writerow()方法写入表头和数据。最后，关闭文件以保存数据。

3. 写入字典数据

除了写入列表数据外，我们还可以写入字典数据到CSV文件。以下是一个示例代码：

# 写入字典数据到CSV文件
data = [
    {'Name': 'David', 'Age': 28, 'City': 'Boston'},
    {'Name': 'Emily', 'Age': 32, 'City': 'Seattle'},
    {'Name': 'Frank', 'Age': 36, 'City': 'Miami'}
]

with open('data_dict.csv', 'w', newline='') as file:
    fieldnames = ['Name', 'Age', 'City']
    writer = csv.DictWriter(file, fieldnames=fieldnames)
    writer.writeheader()
    for row in data:
        writer.writerow(row)

在上面的示例中，我们首先定义一个包含字典数据的列表data，每个字典代表一行数据。然后我们打开一个名为data_dict.csv的文件并指定为写入模式。接着，我们定义字段名称fieldnames和创建一个csv字典写入对象writer，使用writeheader()方法写入表头，并循环写入每一行数据。

4. 运行结果

在上面的示例代码中，我们创建了两个CSV文件data.csv和data_dict.csv并写入了数据。运行代码后，会生成两个CSV文件，并且文件内容如下所示：

data.csv

Name,Age,City
Alice,25,New York
Bob,30,San Francisco
Charlie,35,Chicago

data_dict.csv

Name,Age,City
David,28,Boston
Emily,32,Seattle
Frank,36,Miami

通过上面的示例，我们了解了如何使用Python写入CSV文件，并且可以根据自己的需求调整数据和字段。CSV文件是一种常见的数据交换格式，可以方便地在不同系统之间进行数据传输和共享。