pip install pandas openpyxl

import os
import pandas as pd

def split_excel_by_column(input_file, split_column, output_dir="拆分結(jié)果"):
    """
    按指定列的值將 Excel 總表拆分為多個獨立文件

    參數(shù):
        input_file: 輸入的總表 Excel 文件路徑
        split_column: 用于拆分的列名（如"部門"、"城市"）
        output_dir: 輸出文件夾路徑
    """
    # 1. 讀取總表
    if not os.path.exists(input_file):
        print(f"錯誤：文件 {input_file} 不存在！")
        return

    df = pd.read_excel(input_file, engine='openpyxl')
    print(f"總表數(shù)據(jù)：{len(df)} 行，{len(df.columns)} 列")

    # 2. 檢查拆分列是否存在
    if split_column not in df.columns:
        print(f"錯誤：列名 '{split_column}' 不存在！可用列：{list(df.columns)}")
        return

    # 3. 創(chuàng)建輸出目錄
    os.makedirs(output_dir, exist_ok=True)

    # 4. 按指定列分組并導(dǎo)出
    grouped = df.groupby(split_column)
    group_count = 0

    for group_name, group_df in grouped:
        # 清理文件名中的非法字符（Windows 文件名不允許的字符）
        safe_name = str(group_name).replace('/', '_').replace('\\', '_')
        safe_name = safe_name.replace('*', '').replace('?', '').replace(':', '')

        output_file = os.path.join(output_dir, f"{safe_name}.xlsx")
        group_df.to_excel(output_file, index=False, engine='openpyxl')
        group_count += 1
        print(f"[{group_count}] 已拆分: {safe_name} ({len(group_df)} 行)")

    print(f"\n拆分完成！共生成 {group_count} 個文件，保存在: {output_dir}/")


# ==================== 使用示例 ====================
if __name__ == "__main__":
    # ========== 示例 1：按部門拆分 ==========
    split_excel_by_column(
        input_file="員工花名冊總表.xlsx",
        split_column="部門",
        output_dir="按部門拆分"
    )

    # ========== 示例 2：按城市拆分 ==========
    # split_excel_by_column(
    #     input_file="銷售數(shù)據(jù)總表.xlsx",
    #     split_column="城市",
    #     output_dir="按城市拆分"
    # )

df = pd.read_excel(input_file, engine='openpyxl')

if split_column not in df.columns:
    print(f"錯誤：列名 '{split_column}' 不存在！可用列：{list(df.columns)}")
    return

grouped = df.groupby(split_column)

for group_name, group_df in grouped:
    output_file = os.path.join(output_dir, f"{group_name}.xlsx")
    group_df.to_excel(output_file, index=False, engine='openpyxl')

safe_name = str(group_name).replace('/', '_').replace('\\', '_')

df['拆分鍵'] = df['部門'] + '_' + df['城市']
grouped = df.groupby('拆分鍵')

for group_name, group_df in grouped:
    # 只保留需要的列
    keep_columns = ['姓名', '工號', '郵箱', '電話']
    existing_columns = [col for col in keep_columns if col in group_df.columns]
    group_df[existing_columns].to_excel(output_file, index=False, engine='openpyxl')

import shutil

# 拆分完成后打包為 ZIP
shutil.make_archive("拆分結(jié)果", 'zip', output_dir)
print("已打包為 拆分結(jié)果.zip")

# 定義工資區(qū)間
conditions = [
    (df['工資'] < 8000),
    (df['工資'] >= 8000) & (df['工資'] < 15000),
    (df['工資'] >= 15000)
]
choices = ['初級', '中級', '高級']
df['工資等級'] = pd.cut(df['工資'], bins=[0, 8000, 15000, float('inf')], labels=['初級', '中級', '高級'])
grouped = df.groupby('工資等級')

print(df['部門'].unique())  # 查看唯一的部門值
print(df['部門'].value_counts())  # 查看每個部門的行數(shù)

from openpyxl import load_workbook

# 先復(fù)制模板
import shutil
shutil.copy("模板.xlsx", output_file)

# 再寫入數(shù)據(jù)
wb = load_workbook(output_file)
ws = wb.active
for row_idx, row in enumerate(group_df.values, start=2):
    for col_idx, value in enumerate(row, start=1):
        ws.cell(row=row_idx, column=col_idx, value=value)
wb.save(output_file)