import?pandas?as?pd

wumwlmwdm=158520#上層土壤蓄水容量、下層土壤蓄水容量、深層土壤蓄水容量
wm=wum+wlm+wdm#流域土壤平均蓄水容量
wu0wl0wd0=02.220#上層土壤蓄水量初值、下層土壤蓄水量初值、深層土壤蓄水量初值
w0=wu0+wl0+wd0#流域平均土壤蓄水量初值
bkc=0.30.950.14#流域不均勻系數(shù)、蒸發(fā)折算系數(shù)、蒸發(fā)擴散系數(shù)
wmm=wm*（1+b）#最大土壤蓄水容量

filename=‘PEdata.xlsx‘
pedata=pd.read_excel（filenamesheet_name=‘Sheet1‘）#讀取excel數(shù)據(jù)：時間、蒸發(fā)能力、降水
timelen=pedata.shape[0]#時段數(shù)，即數(shù)據(jù)行數(shù)（不包含表頭）
t=list（pedata.iloc[:0]）#日期
ep=list（pedata.iloc[:1]）#日蒸發(fā)能力
p=list（pedata.iloc[:2]）#日降水量

eu=[0]*timelen#上層蒸發(fā)
el=[0]*timelen#下層蒸發(fā)
ed=[0]*timelen#深層蒸發(fā)
e=[0]*timelen#日蒸發(fā)量
pe=[0]*timelen#扣除降水的蒸發(fā)量
dw=[0]*timelen#日土壤蓄水變量
r=[0]*timelen#產(chǎn)流
wu=[0]*（timelen+1）#上層蓄水量
wl=[0]*（timelen+1）#下層蓄水量
wd=[0]*（timelen+1）#深層蓄水量
w=[0]*（timelen+1）#日蓄水量
wu[0]=wu0
wl[0]=wl0
wd[0]=wd0
w[0]=w0

a=[0]*timelen#蓄滿產(chǎn)流模型參數(shù)A
a[0]=wmm*（1-（1-w0/wm）**（1/（1+b）））

for?i?in?range（timelen）:
????????if?wu[i]+p[i]>=ep[i]:#計算eu
????????????????eu[i]=ep[i]
????????else:
????????????????eu[i]=p[i]+wu[i]
????????????????
????????if?p[i]+wu[i]>=ep[i]:#計算el
????????????????el[i]=0
????????else:
????????????????if?wl[i]>=c*wlm:
????????????????????????el[i]=（ep[i]-eu[i]）*wl[i]/wlm
????????????????else:
????????????????????????if?wl[i]>=c*（ep[i]-eu[i]）:
????????????????????????????????el[i]=c*（ep[i]-eu[i]）
????????????????????????else:
????????????????????????????????el[i]=wl[i]
????????
????????if?wu[i]+p[i]????????????????if?wl[i]????????????????????????ed[i]=c*（ep[i]-eu[i]）-el[i]
????????????????else:
???????????????