python 二维插值

在 Python 中实现二维插值，可以使用 SciPy 库中的 interpolate 模块。具体实现步骤如下：

1. 安装 SciPy 库：

pip install scipy

2. 导入 interpolate 模块：

from scipy import interpolate

3. 定义原始数据点的横纵坐标和值：

x = [0, 1, 2]
y = [0, 1, 2]
z = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

4. 使用 interpolate 模块中的 interp2d 函数进行插值：

f = interpolate.interp2d(x, y, z, kind='linear')

interp2d 函数的参数解释：

- x：原始数据点的横坐标数组；
- y：原始数据点的纵坐标数组；
- z：原始数据点的值数组，需要保证与 x 和 y 的形状一致；
- kind：插值方法，可以选择 'linear'、'cubic'、'quintic' 等方法。

5. 调用插值函数 f，输入新的横纵坐标，得到插值后的值：

new_x = [0.5, 1.5, 2.5]
new_y = [0.5, 1.5, 2.5]
new_z = f(new_x, new_y)

interp2d 函数返回的是一个函数对象 f，需要调用 f 才能进行插值。

完整代码如下：

from scipy import interpolate

x = [0, 1, 2]
y = [0, 1, 2]
z = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

f = interpolate.interp2d(x, y, z, kind='linear')

new_x = [0.5, 1.5, 2.5]
new_y = [0.5, 1.5, 2.5]
new_z = f(new_x, new_y)

print(new_z)

输出结果为：

[[ 1.5  3.   4.5]
 [ 4.   5.   6. ]
 [ 6.5  7.   7.5]]