一、引言
将多个DataFrame写入同一个excel文件中,每个DataFrame作为一个sheet,可以使用pandas库中的ExcelWriter类。这个类可以在一个Excel文件中创建多个sheet,并将不同的数据写入这些sheet中。本文演示如何将多个DataFrame写入同一个CSV文件中,每个DataFrame作为一个sheet。
二、实现过程
2.1 代码
import pandas as pd
# 创建多个DataFrame
df1 = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df2 = pd.DataFrame({'C': [7, 8, 9], 'D': [10, 11, 12]})
# 指定要写入的excel文件路径
file_path = 'dataframes.xlsx'
# 创建ExcelWriter对象
with pd.ExcelWriter(file_path) as writer:
# 将每个DataFrame写入不同的sheet
df1.to_excel(writer, sheet_name='Sheet1', index=False)
df2.to_excel(writer, sheet_name='Sheet2', index=False)
print('DataFrames have been written to CSV file successfully.')
2.2 解读
在这个示例中,首先创建了两个DataFrame df1 和 df2。然后我们指定了要写入的CSV文件路径为dataframes.xlsx。接着,我们使用pd.ExcelWriter()来创建一个ExcelWriter对象,然后通过to_excel()方法将每个DataFrame写入到Excel文件中的不同sheet中,分别命名为Sheet1和Sheet2。最后,我们在with块之外打印一条成功消息。
运行上述代码后,你将会在指定的路径下看到一个名为dataframes.xlsx的文件,其中包含了两个sheet,分别对应着df1和df2的数据。
作者简介:
读研期间发表6篇SCI数据挖掘相关论文,现在某研究院从事数据算法相关科研工作,结合自身科研实践经历不定期分享关于Python、机器学习、深度学习、人工智能系列基础知识与应用案例。致力于只做原创,以最简单的方式理解和学习,关注我一起交流成长。需要数据集和源码的小伙伴可以关注底部公众号添加作者微信。