增加通过输入省份单独生成EXCEL
时间: 2024-03-14 08:46:46 浏览: 78
全国省份城市Excel.xlsx
好的,可以考虑修改代码,添加一个输入省份的功能,让用户可以选择生成哪个省份的 Excel 文件。可以按照以下步骤进行修改:
1. 在代码开头添加以下语句,用于获取用户输入的省份名称:
```
province = input("请输入省份名称:")
```
2. 修改筛选条件,只选择用户输入的省份:
```
new_df = data[(data["省份"] == province) & (data["处理情况"] == "未处理")]
```
3. 将生成的 Excel 文件名修改为用户输入的省份名称:
```
new_df.to_excel(province + "旧机具未处理明细.xlsx", sheet_name=province, index=False)
```
修改后的完整代码如下:
```
import pandas as pd
data = pd.read_excel(r'E:\每日数据\MISPOS明细.xlsx', dtype={'商户编号': str, '终端编号': str, '处理情况': str})
rows = data.shape[0] # 获取行数 shape[1]获取列数
province_list = []
for i in range(rows):
temp = data["省份"][i]
if temp not in province_list:
province_list.append(temp) # 将省份的分类存在一个列表中
province = input("请输入省份名称:")
if province in province_list:
new_df = data[(data["省份"] == province) & (data["处理情况"] == "未处理")]
new_df.to_excel(province + "旧机具未处理明细.xlsx", sheet_name=province, index=False)
print("已生成 " + province + " 旧机具未处理明细 Excel 文件!")
else:
print("输入的省份名称不正确,请重新输入!")
```
这样,用户就可以输入自己想要生成的省份名称,程序会自动筛选出该省份的数据,并生成该省份的 Excel 文件。
阅读全文