scipy.interpolate.

UnivariateSpline#

class scipy.interpolate.UnivariateSpline(x, y, w=None, bbox=[None, None], k=3, s=None, ext=0, check_finite=False)[源代码][源代码]#

对给定数据点的1-D平滑样条拟合。

拟合一个度数为 k 的样条曲线 y = spl(x) 到提供的 x, y 数据。s 通过指定平滑条件来确定结点的数量。

参数:

x(N,) 数组类

独立输入数据的1-D数组。必须是递增的；如果 s 为 0，则必须是严格递增的。

y(N,) 数组类

依赖输入数据的1-D数组，长度与 x 相同。

w(N,) 数组类，可选

样条拟合的权重。必须是正数。如果 w 为 None，则所有权重均为 1。默认为 None。

bbox(2,) 数组类，可选

2-序列指定近似区间的边界。如果 bbox 为 None，bbox=[x[0], x[-1]]。默认为 None。

kint, 可选

平滑样条的度数。必须满足 1 <= k <= 5。k = 3 是三次样条。默认值为 3。

s浮点数或无，可选

用于选择节点数量的正平滑因子。节点数量将增加，直到满足平滑条件:

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

然而，由于数值问题，实际条件是:

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

如果 s 是 None，s 将被设置为 len(w)，用于使用所有数据点的平滑样条。如果 s 为 0，样条将插值通过所有数据点。这等同于 InterpolatedUnivariateSpline。默认值是 None。用户可以使用 s 来控制拟合的紧密性和平滑性之间的权衡。较大的 s 意味着更多的平滑，而较小的 s 值表示较少的平滑。推荐的 s 值取决于权重 w。如果权重表示 y 的标准差的倒数，那么一个好的 s 值应该在范围 (m-sqrt(2*m), m+sqrt(2*m)) 内找到，其中 m 是 x、y 和 w 中的数据点数量。这意味着如果 1/w[i] 是 y[i] 标准差的估计值，那么 s = len(w) 应该是一个好的值。

扩展int 或 str, 可选

控制不在由节点序列定义的区间内的元素的外推模式。

如果 ext=0 或 ‘extrapolate’，返回外推值。
如果 ext=1 或 ‘zeros’，返回 0
如果 ext=2 或 ‘raise’，则引发 ValueError
如果 ext=3 或 ‘const’，返回边界值。

默认值为 0。

check_finitebool, 可选

是否检查输入数组是否仅包含有限数字。禁用可能会提高性能，但如果输入包含无穷大或NaN，则可能会导致问题（崩溃、非终止或无意义的结果）。默认值为False。

方法

`__call__`(x[, nu, ext])	在位置 x 处评估样条（或其 nu 阶导数）。
`antiderivative`([n])	构建一个新的样条曲线，表示此样条曲线的反导数。
`derivative`([n])	构造一个新的样条曲线，表示此样条曲线的导数。
`derivatives`(x)	返回在点 x 处样条的所有导数。
`get_coeffs`()	返回样条系数。
`get_knots`()	返回样条曲线的内部节点位置。
`get_residual`()	返回样条近似中残差的加权平方和。
`integral`(a, b)	返回给定两点之间样条的定积分。
`roots`()	返回样条的零点。
`set_smoothing_factor`(s)	使用给定的平滑因子 s 和在上次调用中找到的节点继续样条计算。

validate_input

参见

BivariateSpline: 用于双变量样条的基类。
SmoothBivariateSpline: 通过给定点平滑的双变量样条
LSQBivariateSpline: 使用加权最小二乘法拟合的双变量样条
RectSphereBivariateSpline: 球面上矩形网格上的二元样条
SmoothSphereBivariateSpline: 球坐标系中的平滑二元样条
LSQSphereBivariateSpline: 使用加权最小二乘拟合的球坐标系中的二元样条
RectBivariateSpline: 矩形网格上的二元样条
InterpolatedUnivariateSpline: 给定一组数据点，进行插值的一元样条。
bisplrep: 一个用于找到曲面的双变量B样条表示的函数
bisplev: 一个用于评估双变量B样条及其导数的函数
splrep: 一个用于找到1-D曲线的B样条表示的函数
splev: 一个用于评估B样条或其导数的函数
sproot: 一个用于找到三次B样条根的函数
splint: 一个用于计算B样条在两个给定点之间定积分的函数
spalde: 一个用于评估B样条所有导数的函数

注释

数据点的数量必须大于样条度数 k。

NaN 处理：如果输入数组包含 nan 值，结果将不可用，因为底层样条拟合例程无法处理 nan。一种解决方法是使用零权重表示非数字数据点：

>>> import numpy as np
>>> from scipy.interpolate import UnivariateSpline
>>> x, y = np.array([1, 2, 3, 4]), np.array([1, np.nan, 3, 4])
>>> w = np.isnan(y)
>>> y[w] = 0.
>>> spl = UnivariateSpline(x, y, w=~w)

注意需要将 nan 替换为一个数值（只要相应的权重为零，精确值并不重要。）

参考文献

基于 [1]、[2]、[3] 和 [4] 中描述的算法：

[1]

P. Dierckx, “An algorithm for smoothing, differentiation and integration of experimental data using spline functions”, J.Comp.Appl.Maths 1 (1975) 165-184.

[2]

P. Dierckx, “A fast algorithm for smoothing data on a rectangular grid while using spline functions”, SIAM J.Numer.Anal. 19 (1982) 1286-1304.

[3]

P. Dierckx, “An improved algorithm for curve fitting with spline functions”, report tw54, Dept. Computer Science,K.U. Leuven, 1981.

[4]

P. Dierckx, “Curve and surface fitting with splines”, Monographs on Numerical Analysis, Oxford University Press, 1993.

示例

>>> import numpy as np
>>> import matplotlib.pyplot as plt
>>> from scipy.interpolate import UnivariateSpline
>>> rng = np.random.default_rng()
>>> x = np.linspace(-3, 3, 50)
>>> y = np.exp(-x**2) + 0.1 * rng.standard_normal(50)
>>> plt.plot(x, y, 'ro', ms=5)

使用平滑参数的默认值：

>>> spl = UnivariateSpline(x, y)
>>> xs = np.linspace(-3, 3, 1000)
>>> plt.plot(xs, spl(xs), 'g', lw=3)

手动调整平滑度：

>>> spl.set_smoothing_factor(0.5)
>>> plt.plot(xs, spl(xs), 'b', lw=3)
>>> plt.show()

../../_images/scipy-interpolate-UnivariateSpline-1.png