HPC数据筛选

import pandas as pd
from pathlib import Path
import tkinter.messagebox
import sys
def checkfile():
    path = Path(r'./数据源/HPC.xlsx')
    if path.is_file():
        print('%s正常'%(path))
        print('读取HPC中请稍等...')
    else:
        tkinter.messagebox.showerror(
            title="错误",
            message="%s不存在,请重命名数据源文件为HPC.xlsx拷贝到数据源目录下"%(path))
        sys.exit(0)
checkfile()
#对HPC筛选
HPC = r'./数据源/HPC.xlsx'
df = pd.read_excel(HPC,index_col=None,usecols=['门店','门店名称','商品','销售数量','当前库存'])
while True:
    goodsid = input("请输入商品编码: ")
    try:
        df1 = df[df['商品'] == int(goodsid)]
        print(df1)
        if df1.empty == True:
            print('没有找到%s,请重新输入' % (goodsid))
            continue
        break
    except:
        print('输入的商品编码有误,请重新输入')

class HPC:
    def __init__(self,market,marketname,goods,salenum,stock):
        self.market = market
        self.marketname = marketname
        self.goods = goods
        self.salenum = salenum
        self.stock = stock

    def 销售总数():
        salenumvalues = dflast['销售数量'].values
        salenumSum = 0
        for i in salenumvalues:
            if i * 1 == i:
                salenumSum += i
        print('%s 的销量总数是 %s' % (goodsid, salenumSum))

    def 库存总数():
        stocknumvalues = dflast['当前库存'].values
        stocknumSum = 0
        for i in stocknumvalues:
            if i * 1 == i:
                stocknumSum += i
        print('%s 的库存总数是 %s' % (goodsid, stocknumSum))


#实例化HPC
h = HPC(df1['门店'],df1['门店名称'],df1['商品'],df1['销售数量'],df1['当前库存'])
# 使用DataFrame前将数据整理成字典
dic = {'门店': h.market, '门店名称': h.marketname, '销售数量': h.salenum, '当前库存': h.stock}
dflast = pd.DataFrame(dic)
print(dflast)  # DATAFRAME表最终生成
HPC筛选 = './HPC筛选.xlsx'
# 选择是否根目录生成筛选后的表格
confirminput = input("是否生成筛选后的excel到根目录\n1.生成 2.跳过: ")
if confirminput == "1":
    dflast.to_excel(HPC筛选, index=None, sheet_name=goodsid)
    print('已生成%s' % (HPC筛选))
else:
    print('您选择的是2.跳过')

HPC.库存总数()
HPC.销售总数()
confirminput = input('脚本执行结束，按任意键退出。')