Pandas實(shí)現(xiàn)批量拆分與合并Excel的示例代碼

更新時(shí)間：2022年05月30日 14:28:15 作者：qq_23605533

這篇文章主要為大家詳細(xì)講講如何利用python Pandas實(shí)現(xiàn)批量拆分與合并Excel,文中有非常詳細(xì)的的代碼示例,對(duì)正在學(xué)習(xí)python的小伙伴們很有幫助,需要的朋友可以參考下

前言

提示：這里可以添加本文要記錄的大概內(nèi)容：

將一個(gè)EXCEL等份拆成多個(gè)EXCEL

將多個(gè)小EXCEL合并成一個(gè)大EXCEL并標(biāo)記來(lái)源

提示：以下是本篇文章正文內(nèi)容，下面案例可供參考

一、拆分成小表格

代碼如下（示例）：

import pandas as pd
import os
work_dir=r"G:\360Downloads\myself\zuoye\合并拆分"
splits_dir=f"{work_dir}\splits"
#如果不存在splits文件夾則創(chuàng)建它
if not os.path.exists(splits_dir):
    os.mkdir(splits_dir)
#引入源文件
df_source=pd.read_excel(f"{work_dir}/5月份臺(tái)賬.xlsx",sheet_name="5月份臺(tái)賬87334",skiprows=2)
df_source.head(3)
df_source.index
df_source.shape
total_row_count=df_source.shape[0]
total_row_count
#拆分成多個(gè)大小相同的excel
#1.使用df.iloc方法
#2.使用dataframe.to_excel保存到每個(gè)小excel中
#計(jì)算拆分后小excel的行數(shù)
user_names=["xiaoA","xiaoB","xiaoC","wmy","jzz","xmw"]
#每個(gè)人的任務(wù)數(shù)
split_size=total_row_count//len(user_names)
if total_row_count%len(user_names)!=0:
    split_size+=1
split_size
#拆分多個(gè)dataframe
df_subs=[]
#enumerate() 函數(shù)用于將一個(gè)可遍歷的數(shù)據(jù)對(duì)象(如列表、元組或字符串)組合為一個(gè)索引序列
for idx,user_name in enumerate(user_names):
    #iloc的開(kāi)始索引
    begin=idx*split_size
    #iloc的結(jié)束索引
    end=begin+split_size
    #實(shí)現(xiàn)df按照iloc拆分
    df_sub=df_source.iloc[begin:end]
    #將每個(gè)子df存入列表
    df_subs.append((idx,user_name,df_sub))
#將每個(gè)dataframe存入excel
for idx,user_name,df_sub in df_subs:
    filename=f"{splits_dir}/ee_{idx}_{user_name}.xlsx"
    df_sub.to_excel(filename,index=False)

二、合并excel

1.介紹

1、遍歷文件夾，得到要合并的excel文件列表

2、分別讀取到dataframe，給每一列標(biāo)記來(lái)源

3、使用pd.conca進(jìn)行df批量合并

4、將合并后的dataframe輸出到excel

2.代碼

代碼如下（示例）：

import pandas as pd
import os
work_dir=r"G:\360Downloads\myself\zuoye\合并拆分"
splits_dir=f"{work_dir}\splits"
#如果不存在splits文件夾則創(chuàng)建它
if not os.path.exists(splits_dir):
    os.mkdir(splits_dir)
#遍歷文件夾，得到要合并的excel名稱(chēng)列表
excel_names=[]
for excel_name in os.listdir(splits_dir):
    excel_names.append(excel_name)
excel_names
df_list=[]
for excel_name in excel_names:
    #讀取每個(gè)excel到df
    excel_path=f"{splits_dir}/{excel_name}"
    df_split=pd.read_excel(excel_path)
    username=excel_name.replace("ee_","").replace(".xlsx","")[2:]
    print(excel_name,username)
    #添加列，用戶名字
    df_split["username"]=username
    df_list.append(df_split)
#concat合并
df_merged=pd.concat(df_list)
df_merged.shape
df_merged.head()
df_merged["username"].value_counts()
#輸出
df_merged.to_excel(f"{work_dir}/merged.xlsx",index=False)