pandas 读/写取多个sheet 的excel

经常使用pandas 读取多个sheet 的文件,读取方式,先获得sheet 名字,再指定sheet_name 参数进行读取 ,写多个sheet 到同一个文件 

import pandas as pd

infile = "data/test.xlsx"

raw_file = pd.ExcelFile(infile)
# 获得sheet_names
sheet_names = raw_file.sheet_names

print(sheet_names)
sheet_dict = {}
for sheet in sheet_names:
    df = pd.read_excel(infile,sheet_name=sheet)
    sheet_dict[sheet] = df

outfile = "ret.xlsx"
# 写多个sheet 到 一个excel
with pd.ExcelWriter(outfile, engine="openpyxl") as writer:
    for sheet,sheet_df in sheet_dict.items():
        # 自己添加业务逻辑
        print(sheet_df.shape[0])
        if sheet_df.shape[0] > 1000:
            random_sheet_df = sheet_df.sample(n=1000)
        else:
            random_sheet_df = sheet_df
        all_num +=sheet_df.shape[0]
        sheet_num_dict[sheet] = sheet_df.shape[0]
        random_sheet_df.to_excel(writer,sheet_name=sheet,index=False)

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值