加入可选“省份”,回车输出全部
时间: 2023-10-02 18:08:24 浏览: 37
如果想要让用户有选择地筛选省份,同时也可以选择不输入任何省份进行全部数据的筛选,可以在代码中加入以下代码:
```
# 让用户输入要筛选的省份,如果用户不输入则默认为全部
selected_department = input("请输入要筛选的省份名称(回车输出全部):")
if selected_department == "":
new_df = data[data["处理情况"] == "未处理"]
selected_department = "全部"
else:
new_df = data[(data["省份"] == selected_department) & (data["处理情况"] == "未处理")]
# 将新文件的名称命名为用户输入的省份名称加上一些额外的字符串
new_df.to_excel(selected_department + "旧机具未处理明细.xlsx", sheet_name=selected_department, index=False)
```
这样,用户可以选择输入要筛选的省份名称,如果用户不输入则默认为全部,程序会根据用户输入的名称或者默认的全部,筛选相应的数据,并将结果保存到以省份名称命名的新文件中。
相关问题
回车则按省份逐一输出excel
可以尝试如下代码实现:
```
import pandas as pd
data = pd.read_excel(r'E:\每日数据\MISPOS明细.xlsx', dtype={'商户编号':str, '终端编号':str, '处理情况':str})
department_list = data['省份'].unique().tolist() # 获取所有省份并去重
for department in department_list:
new_df = data[(data['省份'] == department) & (data['处理情况'] == '未处理')] # 筛选出符合要求的数据
new_df.to_excel(str(department) + "旧机具未处理明细" + ".xlsx", sheet_name=department, index=False) # 生成excel文件
print("已完成全部生成excel的操作!")
```
这段代码可以实现按照省份逐一生成对应的excel文件,不需要输入省份名称。
代码中加入输入“省份”名称单独输出文档
如果想要在代码中加入输入“省份”名称单独输出文档的功能,可以对代码进行如下修改:
```
import pandas as pd
import datetime
now = datetime.datetime.now().strftime('%Y%m%d') # 获取当前日期,格式为年月日
data = pd.read_excel(r'E:\每日数据\智能POS明细.xlsx', dtype={'商户编号':str,'终端编号':str})
# 获取所有省份名称
department_list = data['省份'].unique()
# 循环处理每个省份的数据,并输出到对应的文件
for department in department_list:
# 如果输入了省份名称,则只处理该省份的数据
province = input("请输入省份名称,如需处理所有省份请直接回车:")
if province and province != department:
continue
# 使用条件筛选获取该省份的数据
new_df = data[data["省份"] == department]
# 将数据输出到对应的文件
file_name = "{0}智能POS明细{1}.xlsx".format(department, now)
sheet_name = department
new_df.to_excel(file_name, sheet_name=sheet_name, index=False)
```
修改后的代码首先使用 `unique` 方法获取所有省份的名称,并存储在 `department_list` 中。然后,使用一个循环遍历 `department_list` 中的每一个省份,并使用条件筛选获取该省份的数据。在筛选数据时,如果用户输入了省份名称,则只处理该省份的数据;否则,处理所有省份的数据。
最后,将每个省份的数据输出到对应的 Excel 文件中,文件名和工作表名称都使用省份名称来表示。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)