如何将数组传递给 Scipy Interpolate RectBivariateSpline?

How to pass arrays into Scipy Interpolate RectBivariateSpline?

我正在创建一个 Scipy Interpolate RectBivariateSpline 如下:

import numpy as np
from scipy.interpolate import RectBivariateSpline

x = np.array([1,2,3,4])
y = np.array([1,2,3,4,5])
vals = np.array([
    [4,1,4,4,2],
    [4,2,3,2,6],
    [3,7,4,3,5],
    [2,4,5,3,4]
])

rect_B_spline = RectBivariateSpline(x, y, vals)

然后我尝试传入一个 x 和 y 点数组:

a = np.array([3.2, 3.8, 2.2])
b = np.array([2.4, 4.3, 3.3])

print(rect_B_spline(a, b))

我收到如下错误:

Traceback (most recent call last):
  File "path/file", line 18, in <module>
    print(rect_B_spline(a, b))
  File "/path/scipy/interpolate/fitpack2.py", line 728, in __call__
    raise ValueError("Error code returned by bispev: %s" % ier)
ValueError: Error code returned by bispev: 10

当我将 grid=False 参数传递给该方法时,此错误得到纠正。

我对documentation的印象是,如果输入的网格坐标形成规则的网格,那么grid参数应该为True。有什么我想念的吗?

是的,这里的文档可能有点薄弱。您使用的默认调用要求 x 和 y 定义网格点。与创建样条拟合的原始调用一样,这些需要严格按升序排列。然后它将 return 样条插值的完整网格。

如果您改为使用 RectBivariateSpline.ev(xi,yi),在您调用 rect_B_spline.ev(a,b) 的情况下,您将在每个 (xi[0],yi [0]), ..., (xi[j],yi[j]) 数据对.

不太确定您在这里想要哪个 - 如果您想要完整的 x x y 网格,请使每个 x、y 中的点严格升序。如果您想要一系列点的结果,请使用 .ev(x,y) 方法。

这个有效:

RectBivariateSpline(x,y,vals)([2.2,3.2,3.8],[2.4,3.3,4.3],grid=True)

array([[ 3.197056,  2.75356 ,  2.38796 ],
       [ 6.408896,  3.56696 ,  3.46736 ],
       [ 5.768704,  4.33264 ,  3.10824 ]])

RectBivariateSpline(x[:,None],y,vals)(a,b,grid=False)
# array([ 6.408896,  3.10824 ,  2.75356 ])

我按照文档中的说明对值进行了排序:

If grid is True: evaluate spline at the grid points defined by the coordinate arrays x, y. The arrays must be sorted to increasing order.