SciPyでの補間:Yを生成するXを見つける
-
06-07-2019 - |
質問
SciPyで探している Y を提供する X を見つけるより良い方法はありますか? SciPyを使い始めたばかりで、各機能についてあまり詳しくありません。
import numpy as np
import matplotlib.pyplot as plt
from scipy import interpolate
x = [70, 80, 90, 100, 110]
y = [49.7, 80.6, 122.5, 153.8, 163.0]
tck = interpolate.splrep(x,y,s=0)
xnew = np.arange(70,111,1)
ynew = interpolate.splev(xnew,tck,der=0)
plt.plot(x,y,'x',xnew,ynew)
plt.show()
t,c,k=tck
yToFind = 140
print interpolate.sproot((t,c-yToFind,k)) #Lowers the spline at the abscissa
解決
scipyのUnivariateSplineクラスを使用すると、スプラインの処理がよりPythonicになります。
x = [70, 80, 90, 100, 110]
y = [49.7, 80.6, 122.5, 153.8, 163.0]
f = interpolate.UnivariateSpline(x, y, s=0)
xnew = np.arange(70,111,1)
plt.plot(x,y,'x',xnew,f(xnew))
yでxを見つけるには、次のようにします。
yToFind = 140
yreduced = np.array(y) - yToFind
freduced = interpolate.UnivariateSpline(x, yreduced, s=0)
freduced.roots()
xをyで補間することはうまくいくと思ったが、それはやや異なるルートをとる。より多くのポイントでより近いかもしれません。
他のヒント
線形補間のみが必要な場合は、 interp 関数。
あなたの質問を誤解したかもしれませんが、ごめんなさい。 SciPyを使用する必要はないと思います。 NumPyには最小二乗関数があります。
#!/usr/bin/env python
from numpy.linalg.linalg import lstsq
def find_coefficients(data, exponents):
X = tuple((tuple((pow(x,p) for p in exponents)) for (x,y) in data))
y = tuple(((y) for (x,y) in data))
x, resids, rank, s = lstsq(X,y)
return x
if __name__ == "__main__":
data = tuple((
(1.47, 52.21),
(1.50, 53.12),
(1.52, 54.48),
(1.55, 55.84),
(1.57, 57.20),
(1.60, 58.57),
(1.63, 59.93),
(1.65, 61.29),
(1.68, 63.11),
(1.70, 64.47),
(1.73, 66.28),
(1.75, 68.10),
(1.78, 69.92),
(1.80, 72.19),
(1.83, 74.46)
))
print find_coefficients(data, range(3))
これは[128.81280358 -143.16202286 61.96032544]を返します。
>>> x=1.47 # the first of the input data
>>> 128.81280358 + -143.16202286*x + 61.96032544*(x**2)
52.254697219095988
0.04アウト、悪くない
所属していません StackOverflow