【Python数据分析系列】将循环生成的DataFrame写入同一个Excel文件不同工作表（案例+源码）

数据杂坛2024-06-21 10:50

这是我的第306篇原创文章。

一、引言

在做数据分析任务时，我们通常会批量循环处理任务，每个任务可能会产生一个记录信息的dataframe数据表，我们通常需要将每次产生的dataframe数据表保存下来，如何将每次循环产生的dataframe最终保存在一个excel文件的不同数据表里面呢？本文将通过一个简单的案例带给大家答案。

二、实现过程

2.1 案例代码

python 复制代码

import pandas as pd

# 创建一个Excel写入对象
writer = pd.ExcelWriter('output.xlsx', engine='xlsxwriter')

# 模拟循环生成DataFrame的过程
for i in range(5):
    # 在每次循环内生成一个新的DataFrame
    df = pd.DataFrame({'A': [i, i + 1, i + 2], 'B': [i * 2, i * 2 + 1, i * 2 + 2]})

    # 将生成的DataFrame写入Excel文件的不同工作表中
    sheet_name = 'Sheet{}'.format(i + 1)
    df.to_excel(writer, sheet_name=sheet_name)

# 保存Excel文件
writer._save()

2.2 结果

保存为output.xlsx，文件里面有5个sheet表：

作者简介：

读研期间发表6篇SCI数据挖掘相关论文，现在某研究院从事数据算法相关科研工作，结合自身科研实践经历不定期分享关于Python、机器学习、深度学习、人工智能系列基础知识与应用案例。致力于只做原创，以最简单的方式理解和学习，关注我一起交流成长。需要数据集和源码的小伙伴可以关注底部公众号添加作者微信。