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('脚本执行结束,按任意键退出。')
HPC数据筛选
最新推荐文章于 2024-09-20 15:12:07 发布