增加一个input选择按省份生成EXCEL,增加回车后继续按省份自动拆分生成excel
时间: 2024-01-26 07:05:16 浏览: 67
好的,我可以为您提供一个代码示例,实现根据用户输入的省份名称,将数据按照省份拆分,生成单独的Excel文件。您可以将以下代码添加到您的程序中,并根据您的实际情况进行修改:
```
import pandas as pd
# 读取Excel文件
data = pd.read_excel(r'E:\每日数据\MISPOS明细.xlsx', dtype={'商户编号': str, '终端编号': str, '处理情况': str})
# 获取省份列表
department_list = data['省份'].unique()
# 用户输入省份名称
department_name = input('请输入省份名称:')
# 按照省份拆分数据
if department_name in department_list:
new_df = data[data['省份'] == department_name]
file_name = department_name + '旧机具未处理明细.xlsx'
new_df.to_excel(file_name, index=False)
print('已成功生成Excel文件:' + file_name)
else:
print('无法找到该省份!')
# 循环按照省份拆分数据
for department in department_list:
new_df = data[data['省份'] == department]
file_name = department + '旧机具未处理明细.xlsx'
new_df.to_excel(file_name, index=False)
print('已成功生成Excel文件:' + file_name)
```
这段代码首先读取了一个名为`MISPOS明细.xlsx`的Excel文件,并将数据存储在名为`data`的DataFrame对象中。然后,使用`unique()`方法获取了数据中所有不重复的省份,并将这些省份存储在一个名为`department_list`的列表中。
接着,程序会提示用户输入省份名称,如果用户输入的省份名称在`department_list`中,则程序会将数据按照该省份拆分,并将拆分后的数据写入一个名为“省份名称+旧机具未处理明细.xlsx”的Excel文件中。如果用户输入的省份名称不在`department_list`中,则程序会提示“无法找到该省份!”的信息。
最后,程序会循环遍历`department_list`列表中的每个省份,并将数据按照省份拆分,并将拆分后的数据分别写入名为“省份名称+旧机具未处理明细.xlsx”的Excel文件中。在循环过程中,程序会输出“已成功生成Excel文件:省份名称+旧机具未处理明细.xlsx”的信息。
阅读全文