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

一、引言

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

二、实现过程

2.1 案例代码

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、机器学习、深度学习、人工智能系列基础知识与应用案例。致力于只做原创,以最简单的方式理解和学习,关注我一起交流成长。需要数据集和源码的小伙伴可以关注底部公众号添加作者微信。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐