Pandas批量拆分Excel与合并Excel

Pandas批量拆分Excel与合并Excel

  • 将一个大Excel等份拆成多个Excel
  • 合并多个小Excel到一个大Excel

F盘一个Excel表格,具体路径是 F:\Temp\data\work.xlsx。里面记录了三百多个任务,现在要将这些任务相对平均的分到六个人身上。

第一部分。源文件work.xlsx,拆分成六部分。

第二部分。将六个文件合并成一个大文件,work_user.xlsx。

data文件夹 splits文件夹
Pandas批量拆分Excel与合并Excel_第1张图片 Pandas批量拆分Excel与合并Excel_第2张图片

将一个大Excel等份拆成多个Excel

work_dir="F:/Temp/data"
splits_dir=f"{work_dir}/splits"

# 判断工作目录下有没有一个名为 splits 的文件夹。如果没有就创建一个,有就不动
import os
if not os.path.exists(splits_dir):
    os.mkdir(splits_dir)

import pandas as pd
df_source = pd.read_excel(f'{work_dir}/work.xlsx')
total_row_count = df_source.shape[0]

# 这个大excel,会拆分给这几个人
user_names = ["xiao_shuai", "xiao_wang", "xiao_ming", "xiao_lei", "xiao_bo", "xiao_hong"]
# 每个人的任务数目
split_size = total_row_count // len(user_names)
if total_row_count % len(user_names) != 0:
    split_size += 1

# 拆分成多个dataframe
df_subs = []
for idx, user_name in enumerate(user_names):
    # iloc的开始索引
    begin = idx*split_size
    # iloc的结束索引
    end = begin+split_size
    # 实现df按照iloc拆分
    df_sub = df_source.iloc[begin:end]
    # 将每个子df存入列表
    df_subs.append((idx, user_name, df_sub))


# 将每个datafame存入excel
for idx, user_name, df_sub in df_subs:
    file_name = f"{splits_dir}/work_{idx}_{user_name}.xlsx"
    df_sub.to_excel(file_name, index=False)

合并多个小Excel到一个大Excel

  1. 遍历文件夹,得到要合并的Excel文件列表
  2. 分别读取到dataframe,给每个df添加一列用于标记来源
  3. 使用pd.concat进行df批量合并
  4. 将合并后的dataframe输出到excel
work_dir="F:/Temp/data"
splits_dir=f"{work_dir}/splits"

# 遍历文件夹,得到要合并的Excel名称列表
import os
excel_names = []
for excel_name in os.listdir(splits_dir):
    excel_names.append(excel_name)

# 分别读取到dataframe
import pandas as pd

df_list = []

for excel_name in excel_names:
    # 读取每个excel到df
    excel_path = f"{splits_dir}/{excel_name}"
    df_split = pd.read_excel(excel_path)
    # 得到username
    username = excel_name.replace("work_", "").replace(".xlsx", "")[2:]
    
    # 给每个df添加1列,即用户名字
    df_split["username"] = username
    df_list.append(df_split)


# 使用pd.concat进行合并
df_merged = pd.concat(df_list)

# 将合并后的dataframe输出到excel
df_merged.to_excel(f"{work_dir}/work_user.xlsx", index=False)

你可能感兴趣的:(#,Pandas,python)