Sponsored Link

カテゴリー別アーカイブ: scipy

(48) スプライン補間で内挿

(48) スプライン補間で内挿

以下の4点を通る滑らかなそれっぽい曲線を描きたい。
(x,y)=(-2,4),(-1,1),(1,5),(2,1)

この4点を直線で結んでみる。

import numpy as np
import matplotlib.pyplot as plt
x = np.array([-2,-1,1,2],dtype='float32')
y = np.array([4,1,5,1],  dtype='float32')
plt.plot(x, y)
plt.show()

20140829_02

この4点のX軸上の両端を0.1ステップで40分割し、各X座標に対応するY座標を スプライン補間 で求める。

import scipy.interpolate
xn = np.arange(-2.0, 2.1, 0.1)
rp = scipy.interpolate.splrep(x, y, s=0)
yn = scipy.interpolate.splev(xn, rp, der=0)
plt.plot(x, y)
plt.plot(xn, yn)
plt.show()

20140829_03

P.S.
スプライン補間は3次元でも使える。