水库水位-库容曲线线性插值python程序

水位-库容曲线是进行水库调度的基本数据资料。具体用途就是，给定一个水库水位z，求其对应的库容v，或者给定一个库容v，求其对应的水位z。比较靠谱的方法就是线性插值。以下给出一个本人自定义的进行水库水位库容曲线线性插值的python函数。

def Linterpo(x,y,xi):#参数一组升序排列的长度相等的x列表和y列表，以及给定一个x值
    if xi<x[0]:
        yi=y[0]+(xi-x[0])/(x[0]-x[1])*(y[0]-y[1])
    if xi>x[-1]:
        yi=y[-1]+(xi-x[-1])/(x[-1]-x[-2])*(y[-1]-y[-2])
    else:
        for j in range(len(x)):
            if xi>=x[j] and xi<x[j+1]:
                yi=y[j]+(xi-x[j])/(x[j+1]-x[j])*(y[j+1]-y[j])
    return yi
#以下举例验证了函数的正确性
z=[i for i in range(100,110)]#水位列表，单位m
v=[1,1.1,1.21,1.32,1.44,1.57,1.7,1.84,2,2.2]#库容列表，单位亿m^3
vi=Linterpo(z,v,103.4)#由水位求库容
print(vi)
zi=Linterpo(v,z,1.5)#由库容求水位
print(zi)