沿 Z 轴内插 3D 曲面以外推新曲面

Question

我正在尝试插入三个 3D 曲面以沿 Z 轴外推其他 3D 曲面。

在下面的代码中，我展示了我的三个表面。 我需要的是 1）在它们之间创建 N 个曲面，以及 2）可能在第一个和最后一个曲面的下方和上方外推其他曲面。

我尝试使用RegularGridInterpolator，但我只能得出一些观点。 有没有办法获得整个表面（点集），而不是只有一个点？

import numpy as np
import matplotlib.pyplot as plt
from scipy import interpolate

mat1 = np.array([     [60, 50, 40, 30],
                      [55, 45, 35, 25],
                      [44, 34, 24, 14],
                      [38, 28, 18,  8] ])

mat2 = mat1 + 20

mat3 = mat1 + 40

x = np.linspace(0,1,4)
y = np.linspace(0,100,4)
tot = np.array([mat1, mat2, mat3])

x2 = np.linspace(0,1,3)
y2 = np.linspace(0,100,4)
z2 = np.linspace(0,100,4)

fig, ax = plt.subplots(figsize=(6,6),subplot_kw=dict(projection='3d'))

# plot orgiginal surfaces
for zg in tot:
    xg, yg = np.meshgrid(x, y)
    s = ax.plot_surface(xg, yg, zg, alpha=0.8)

# grid interpolator
xg2, yg2, zg2 = np.meshgrid(x2, y2, z2)
interp = interpolate.RegularGridInterpolator((x2, y2, z2), tot)

print(interp([0.5,50,50])) # this gives me only one point z-point in x=0.5 and y=50