python三维线性插值_[Python] 分段线性插值

利用线性函数做插值

每一段的线性函数：

#Program 0.6 Linear Interploation

import numpy as np

import matplotlib.pyplot as plt

#分段线性插值闭包

def get_line(xn, yn):

def line(x):

index = -1

#找出x所在的区间

for i in range(1, len(xn)):

if x <= xn[i]:

index = i-1

break

else:

i += 1

if index == -1:

return -100

#插值

result = (x-xn[index+1])*yn[index]/float((xn[index]-xn[index+1])) + (x-xn[index])*yn[index+1]/float((xn[index+1]-xn[index]))

return result

return line

xn = [i for i in range(-50,50,10)]

yn = [i**2 for i in xn]

#分段线性插值函数

lin = get_line(xn, yn)

x = [i for i in range(-50, 40)]

y = [lin(i) for i in x]

#画图

plt.plot(xn, yn, 'ro')

plt.plot(x, y, 'b-')

plt.show()