对于一个脚本一次会生成多个结果,将多个结果分别导出成多个文件,这类需求相对很简单,就不做概述。我们的需求情形如下:
实际应用中,我们也常会碰到,需要将python处理的多个结果(常常是DataFrame格式存在)保存在同一个excel的不同sheet页中。基于此,python提供了很好的实现方法。具体如下。
pandas提供了ExcelWriter()函数,可以很好的将我们上述的需求分别导出到同一个excel的不同Sheet中。具体示例如下:
import pandas as pd
# 示例数据
df_one = pd.DataFrame([[1,2,3],[4,3,5]])
df_two = pd.DataFrame([['a','b','c'],['k','d','d']])
df_three = pd.DataFrame(['beijing','shanghai','tianjin'])
#声明一个读写对象
writer = pd.ExcelWriter(r'data/数据导入多个sheet页测试.xls',engine='xlsxwriter')
#分别将表df_one、df_two、df_three写入Excel中的Sheet1、Sheet2、Sheet3
# 并命名为表1、表2、表3
df_one.to_excel(writer,sheet_name='表1')
df_two.to_excel(writer,sheet_name='表2')
df_three.to_excel(writer,sheet_name='表3')
#保存读写的内容
writer.save()
上述示例主要,一定要将读写对象做保存操作,即:writer.save() 语句。否则本地会没有结果文件。这也是时常会忘记的操作。
因此,可以借助with语句,防止上述漏写的操作。对应完整代码如下:
import pandas as pd
# 示例数据
df_one = pd.DataFrame([[1,2,3],[4,3,5]])
df_two = pd.DataFrame([['a','b','c'],['k','d','d']])
df_three = pd.DataFrame(['beijing','shanghai','tianjin'])
#声明一个读写对象
with pd.ExcelWriter(r'data/数据导入多个sheet页测试.xls',engine='xlsxwriter') as writer:
df_one.to_excel(writer,sheet_name='表1')
df_two.to_excel(writer,sheet_name='表2')
df_three.to_excel(writer,sheet_name='表3')
上述源码和测试文件结果,想全部获取的可以扫码关注微信公众号,回复关键字【sheet】,来获取下载。