Python 写入 CSV 文件

在数据分析和处理方面，CSV 文件非常常见。Python 提供了多种方法将数据写入 CSV 文件中。这篇文章将展示一些基本的方法来进行 CSV 文件写入。

1. 使用 csv 模块

Python 的 csv 模块提供了用于读取和写入 CSV 文件的功能。

import csv

# 数据列表
data = [
	('John', 'Doe', 'johndoe@example.com'),
	('Jane', 'Doe', 'janedoe@example.com'),
	('Bob', 'Smith', 'bobsmith@example.com'),
]

# 写入 CSV 文件
with open('example.csv', 'w', newline='') as file:
	writer = csv.writer(file)
	writer.writerows(data)

在这个例子中，我们使用 writerows() 方法将数据写入 CSV 文件中。注意文件打开模式为写模式 'w'。

2. 使用 csv.DictWriter

如果您想要写入一个包含键值对的字典列表，您可以使用 csv.DictWriter。

import csv

# 数据列表
data = [
    {'first_name': 'John', 'last_name': 'Doe', 'email': 'johndoe@example.com'},
    {'first_name': 'Jane', 'last_name': 'Doe', 'email': 'janedoe@example.com'},
    {'first_name': 'Bob', 'last_name': 'Smith', 'email': 'bobsmith@example.com'},
]

# 写入 CSV 文件
with open('example.csv', 'w', newline='') as file:
    fieldnames = ['first_name', 'last_name', 'email']
    writer = csv.DictWriter(file, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerows(data)

在这个例子中，我们首先定义了一个包含字典的列表。然后我们创建了一个 csv.DictWriter 对象，指定了要写入的文件名和字段名。在调用 writerows() 方法之前，我们必须先调用 writeheader() 方法来写入 CSV 文件头。

3. 使用 pandas 库

另一个可以用来写入 CSV 文件的库是 pandas。pandas 是一个非常强大的数据分析工具，它提供了许多在数据分析和处理方面非常有用的功能。下面是一个使用 pandas 写入 CSV 文件的例子。

import pandas as pd

# 数据列表
data = [
    {'first_name': 'John', 'last_name': 'Doe', 'email': 'johndoe@example.com'},
    {'first_name': 'Jane', 'last_name': 'Doe', 'email': 'janedoe@example.com'},
    {'first_name': 'Bob', 'last_name': 'Smith', 'email': 'bobsmith@example.com'},
]

# 创建 data frame 对象
df = pd.DataFrame(data)

# 写入 CSV 文件
df.to_csv('example.csv', index=False)

在这个例子中，我们首先创建了一个包含字典的列表。然后我们使用 pandas 创建了一个 data frame 对象。最后，我们使用 to_csv() 方法将 data frame 对象写入 CSV 文件中。

以上是 Python 写入 CSV 文件的三种简单方法。根据实际用途选择适合您的方法吧！