# 步骤一：读取表格
import pandas as pd

# dirPath = input('请输入源文件地址:')
dirPath = 'C:\\tmp\\'
ym = '202411'
monthname = '2024年11月'
path = dirPath + 'data_shengtk_' + ym + 'zz.xlsx'

orgName = pd.read_excel(path)
org_list = list(orgName['分区'].drop_duplicates())
print(org_list)

# 步骤二：完成表格的拆分  表1 xx县-x月-低保导入模版（在保家庭及人员）.xls
for i2 in org_list:
    print(i2)
    cname = '其他'
    i = str(i2)

     # 使用字典简化条件判断
    cname_map = {
        '220202': '昌邑区',
        '220203': '龙潭区',
        '220204': '船营区',
        '220211': '丰满区',
        '220292': '高新区',
        '220291': '经开区'
    }
    cname = cname_map.get(i, '其他')

    print(cname)

    writer = pd.ExcelWriter(dirPath + cname + '-' + monthname + '-特困导入模板（在保家庭及人员全部）.xlsx',
                            engine='xlsxwriter')
    tempdata = orgName[orgName['分区'] == int(i)]
    # 删除分区这列
    data = tempdata.drop(columns='分区')
    # 首列插入序号
    data.insert(0, '序号', range(1, 1 + len(data)))
    data[['村/社区区划编码', '家庭编号']] = data[['村/社区区划编码', '家庭编号']].astype(str)
    data.to_excel(writer, index=False)
    writer.close()