Python循环将DataFrame写入CSV

在数据分析和处理中，我们常常需要将数据保存到文件中，以便后续使用或者与他人共享。而CSV（逗号分隔值）是一种常见的文件格式，它可以简单地将数据以文本形式保存，每个值用逗号进行分隔。

本文将介绍如何使用Python循环将DataFrame写入CSV文件，并提供代码示例。首先，我们需要导入必要的库。

import pandas as pd

准备数据

在开始之前，我们需要准备一些数据作为示例。假设我们有一个包含学生信息的DataFrame，包括学生姓名、年龄和成绩。

data = {'姓名': ['张三', '李四', '王五'],
        '年龄': [18, 19, 20],
        '成绩': [90, 85, 95]}

df = pd.DataFrame(data)

我们可以使用pd.DataFrame()函数将字典类型的数据转换为DataFrame。

将DataFrame写入CSV文件

现在我们已经有了一个包含学生信息的DataFrame，接下来我们将其写入CSV文件。

df.to_csv('students.csv', index=False)

这里使用了to_csv()函数，它将DataFrame写入CSV文件。我们指定了文件名为"students.csv"，并通过index=False参数禁止写入行索引。

如果你打开"students.csv"文件，你将看到以下内容：

姓名,年龄,成绩
张三,18,90
李四,19,85
王五,20,95

循环写入多个DataFrame

有时候，我们可能需要循环将多个DataFrame写入不同的CSV文件。我们可以使用循环来实现这个目标。

data1 = {'姓名': ['张三', '李四', '王五'],
         '年龄': [18, 19, 20],
         '成绩': [90, 85, 95]}

data2 = {'姓名': ['赵六', '钱七', '孙八'],
         '年龄': [21, 22, 23],
         '成绩': [80, 75, 85]}

dfs = [pd.DataFrame(data1), pd.DataFrame(data2)]
file_names = ['students1.csv', 'students2.csv']

for df, file_name in zip(dfs, file_names):
    df.to_csv(file_name, index=False)

在以上代码中，我们首先定义了两个DataFrame（data1和data2），然后将它们存储在一个列表中。我们还定义了一个包含两个文件名的列表（file_names）。

接下来，我们使用zip()函数将DataFrame和文件名进行配对，然后通过循环遍历这些配对，并将每个DataFrame写入对应的CSV文件。

总结

在本文中，我们介绍了如何使用Python循环将DataFrame写入CSV文件。我们首先导入了必要的库，然后准备了一个包含学生信息的DataFrame作为示例。我们使用了to_csv()函数将DataFrame写入CSV文件，并演示了如何循环写入多个DataFrame。

希望本文对你了解如何将DataFrame写入CSV文件有所帮助！