Python實現(xiàn)線性插值和三次樣條插值的示例代碼

更新時間：2019年11月13日 09:52:21 作者：XHHP

這篇文章主要介紹了Python實現(xiàn)線性插值和三次樣條插值的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

(1)、函數(shù)

y = sin(x)

(2)、數(shù)據(jù)準備

#數(shù)據(jù)準備
X=np.arange(-np.pi,np.pi,1) #定義樣本點X，從-pi到pi每次間隔1
Y= np.sin(X)#定義樣本點Y，形成sin函數(shù)
new_x=np.arange(-np.pi,np.pi,0.1) #定義差值點

(3)、樣條插值

#進行樣條差值
import scipy.interpolate as spi
 
#進行一階樣條插值
ipo1=spi.splrep(X,Y,k=1) #樣本點導入，生成參數(shù)
iy1=spi.splev(new_x,ipo1) #根據(jù)觀測點和樣條參數(shù)，生成插值
 
#進行三次樣條擬合
ipo3=spi.splrep(X,Y,k=3) #樣本點導入，生成參數(shù)
iy3=spi.splev(new_x,ipo3) #根據(jù)觀測點和樣條參數(shù)，生成插值

(4)、繪圖

##作圖
fig,(ax1,ax2)=plt.subplots(2,1,figsize=(10,12))

ax1.plot(X,Y,'o',label='樣本點')
ax1.plot(new_x,iy1,label='插值點')
ax1.set_ylim(Y.min()-1,Y.max()+1)
ax1.set_ylabel('指數(shù)')
ax1.set_title('線性插值')
ax1.legend()

ax2.plot(X,Y,'o',label='樣本點')
ax2.plot(new_x,iy3,label='插值點')
ax2.set_ylim(Y.min()-1,Y.max()+1)
ax2.set_ylabel('指數(shù)')
ax2.set_title('三次樣條插值')
ax2.legend()

(5)、綜合代碼

import numpy as np
import matplotlib.pyplot as plt
 
plt.rcParams['font.sans-serif']=['SimHei'] #用來正常顯示中文標簽
plt.rcParams['axes.unicode_minus']=False #用來正常顯示負號
 
#數(shù)據(jù)準備
X=np.arange(-np.pi,np.pi,1) #定義樣本點X，從-pi到pi每次間隔1
Y= np.sin(X)#定義樣本點Y，形成sin函數(shù)
new_x=np.arange(-np.pi,np.pi,0.1) #定義差值點
 
#進行樣條差值
import scipy.interpolate as spi
 
#進行一階樣條插值
ipo1=spi.splrep(X,Y,k=1) #樣本點導入，生成參數(shù)
iy1=spi.splev(new_x,ipo1) #根據(jù)觀測點和樣條參數(shù)，生成插值
 
#進行三次樣條擬合
ipo3=spi.splrep(X,Y,k=3) #樣本點導入，生成參數(shù)
iy3=spi.splev(new_x,ipo3) #根據(jù)觀測點和樣條參數(shù)，生成插值

 
 
##作圖
fig,(ax1,ax2)=plt.subplots(2,1,figsize=(10,12))

ax1.plot(X,Y,'o',label='樣本點')
ax1.plot(new_x,iy1,label='插值點')
ax1.set_ylim(Y.min()-1,Y.max()+1)
ax1.set_ylabel('指數(shù)')
ax1.set_title('線性插值')
ax1.legend()

ax2.plot(X,Y,'o',label='樣本點')
ax2.plot(new_x,iy3,label='插值點')
ax2.set_ylim(Y.min()-1,Y.max()+1)
ax2.set_ylabel('指數(shù)')
ax2.set_title('三次樣條插值')
ax2.legend()

(6)、結果展示