python樣條插值的實現(xiàn)代碼

發(fā)布時間：2020-08-23 16:15:30 來源：腳本之家閱讀：179 作者：lucindawuyi 欄目：開發(fā)技術(shù)

本文實例為大家分享了python樣條插值的具體實現(xiàn)代碼，供大家參考，具體內(nèi)容如下

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
 
plt.rcParams['font.sans-serif']=['SimHei'] #用來正常顯示中文標簽
plt.rcParams['axes.unicode_minus']=False #用來正常顯示負號
 
#導入數(shù)據(jù)
data1=pd.read_csv('data1.csv',encoding='gbk')
 
#數(shù)據(jù)準備
X=data1.index #定義數(shù)據(jù)點
Y=data1['滬深300'].values #定義數(shù)據(jù)點
x=np.arange(0,len(data1),0.15) #定義觀測點
 
#進行樣條差值
import scipy.interpolate as spi
 
#進行一階樣條差值
ipo1=spi.splrep(X,Y,k=1) #源數(shù)據(jù)點導入，生成參數(shù)
iy1=spi.splev(x,ipo1) #根據(jù)觀測點和樣條參數(shù)，生成插值
 
#進行三次樣條擬合
ipo3=spi.splrep(X,Y,k=3) #源數(shù)據(jù)點導入，生成參數(shù)
iy3=spi.splev(x,ipo3) #根據(jù)觀測點和樣條參數(shù)，生成插值
 
 
##作圖
fig,(ax1,ax2)=plt.subplots(2,1,figsize=(10,12))
ax1.plot(X,Y,label='滬深300')
ax1.plot(x,iy1,'r.',label='插值點')
ax1.set_ylim(Y.min()-10,Y.max()+10)
ax1.set_ylabel('指數(shù)')
ax1.set_title('線性插值')
ax1.legend()
ax2.plot(X,Y,label='滬深300')
ax2.plot(x,iy3,'b.',label='插值點')
ax2.set_ylim(Y.min()-10,Y.max()+10)
ax2.set_ylabel('指數(shù)')
ax2.set_title('三次樣條插值')
ax2.legend()

基于《Python 與量化投資從基礎(chǔ)到實戰(zhàn)》的內(nèi)容練習

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持億速云。

向AI問一下細節(jié)

python樣條插值的實現(xiàn)代碼

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標簽