如何为此数据指定最佳样条

时间:2018-12-11 17:01:40

标签: python scipy spline

import numpy as np
import matplotlib.pyplot as plt
from scipy.interpolate import UnivariateSpline
x = [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47]
y = [31031.4,35241.7,42320.8,38146.7,38345.9,42320.8,33649.2,42883.3,35459,36371.6,23259.8,53740.6,36971.6,29518.3,45773,51421.43,56303.5,38786,31956.4,42717,43026.88,58479.4,50628.24,33375.19,40332.36,52731.29,48409.99,62328.65,48409.99,53941.38,52736.68,55794.38,48180.88,42949.35,48234.66,49505.83,50629.87,25361.3,57403.21,46805.93,55834.61,24978.13,51037.43,46727.27,31766.46,21553.01,34017.05,31533.17]
plt.plot(x, y, 'ro', ms=5)
spl = UnivariateSpline(x, y)
print(spl.get_knots())
xs = np.linspace(0, 47, 1000)
plt.plot(xs, spl(xs), 'g', lw=3)
#spl.set_smoothing_factor(1000000000)
#plt.plot(xs, spl(xs), 'b', lw=3)
plt.show()

我具有样条曲线遍历所有点的图形(屏幕1),但是我需要像屏幕2上的图形。
屏幕1:

屏幕2:

1 个答案:

答案 0 :(得分:1)

UnivariateSpline具有一个平滑参数s,请参见here

  

用于选择结数的正平滑因子。结数将增加,直到满足平滑条件为止:

     

sum((w[i] * (y[i]-spl(x[i])))**2, axis=0) <= s

由于y值很大,因此平滑因子也必须很大(否则您可以对数据进行归一化,请参见下文)。

spl = UnivariateSpline(x, y, s=1e9)

看起来像这样:

enter image description here

标准化:

spl = UnivariateSpline(x, y/np.max(y))
xs = np.linspace(0, 47, 1000)
plt.plot(xs, spl(xs)*np.max(y), 'g', lw=3)

s = len(w)(请参阅链接,w是权重),这是默认值。 情节:

enter image description here

spl = UnivariateSpline(x, y/np.max(y), s=float(len(y))*0.017)
xs = np.linspace(0, 47, 1000)
plt.plot(xs, spl(xs)*np.max(y), 'g', lw=3)

外观与您的情节相似:

enter image description here