在 python 中使用来自不同大小网格点的数据生成 3-D 曲面图

Question

我从this 问题中学会了在 python 中合并来自两个矩形的网格网格点。合并后的网格点为：

这里每个网格点都包含它们各自的值，我想绘制它的 3-D 曲面图。对于完整的网格（没有空白点），我会编写以下代码：

fig = plt.figure()
ax = fig.add_subplot(111, projection = '3d')
x=np.arange(0,10,1)
y=np.arange(0,12,1)
X, Y = np.meshgrid(x,y)
print (Y)
Z = gridValue.reshape(X.shape)
cset = ax.pcolormesh(X, Y, Z, cmap=cm.coolwarm, linewidth =0, antialiased = False)
ax.clabel(cset, fontsize=9, inline=1)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('σ ')
fig.colorbar(cset, shrink=0.5, aspect =5)
plt.show()

当我尝试用上面的代码绘制网格值（80 个点）时，我得到一个明显的错误：

ValueError: cannot reshape array of size 80 into shape (12,10)

我期待下图（不完全但类似的方式）作为最终的 3-D 曲面图：

我的情况如何绘制 3-D 曲面图？

谢谢。

编辑：我尝试使用以下代码 sn-p 进行绘图：

fig = plt.figure()
ax = fig.add_subplot(111, projection = '3d')
x1=np.arange(0,2,1)
x2 = np.arange(0,10,1)
y1 =np.arange(0,5,1)
y2=np.arange(5,12,1)
Y1, X1 = np.meshgrid(x1,y1)
Y2, X2 = np.meshgrid(x2,y2)
gridValue1 = gridValue[:10]
gridValue2 = gridValue[10:]
Z1 = gridValue1.reshape(X1.shape)
Z2 = gridValue2.reshape(X2.shape)
cset1 = ax.plot_surface(X1, Y1, Z1, cmap=cm.coolwarm, linewidth =0, antialiased = False)
cset2 = ax.plot_surface(X2, Y2, Z2, cmap=cm.coolwarm, linewidth =0, antialiased = False)
ax.clabel(cset1, fontsize=9, inline=1)
ax.clabel(cset2, fontsize=9, inline=1)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('σ ')
plt.show()

我用这段代码得到了下图：

现在的问题是说明这个数字。颜色代码（红色表示高值，蓝色表示低值）不再有效。因为蓝色网格点（第一张图中）的gridValues 很高，所以大部分在曲面图中应该是红色的。因此，合并两个单独的 3-D 曲面图并不是解决我的问题的方法。任何帮助，请！

Answer 1

如果您的gridValue 数组大于带有(12,10) 的网格，您需要将gridValue 插值到正确的形状。 在你的情况下，因为我猜你想用80点绘制显示的形状，你需要为gridValue构造一个具有meshgrid形状的数组，然后将你的80个网格值分配到正确的位置gridValue 形状为 (12,10) 的数组。您如何进行分发完全取决于您拥有的信息，因此不幸的是我无法帮助您。但例如考虑蓝色和绿色网格的 y 排序顺序分布：

gridValue_new = np.zeros_like(X)
gridValue_new[:, 0] = gridValue[:12]
gridValue_new[:, 1] = gridValue[12:24]
gridValue_new.reshape(-1)[:8, 2:] = gridValue[24:]

前两个 ligns 分布点也可以使用reshape() 重写。

如果你想在 2d 中绘制它，请使用 pcolormesh。 对于 3d 绘图，使用 ax.plot_surface() 将满足您的需求。 如果要在一个轴上绘制两个网格，请执行以下操作：

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt

X, Y = np.meshgrid(np.arange(0, 10, 1), np.arange(0, 12, 1))
X1 = X[5:, :]
Y1 = Y[5:, :]
X2 = X[0:5, 0:2]
Y2 = Y[0:5, 0:2]
Z1 = np.random.rand(7, 10)  # generating some random grid values
Z2 = np.random.rand(5, 2)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(X1, Y1, Z1)
ax.plot_surface(X2, Y2, Z2)

要获得正确的颜色条，请获取数组的最小值/最大值：

vmin = gridValue.min()
vmax = gridValue.max()

并将两者都作为参数传递给plot_surface():

cset1 = ax.plot_surface(X1, Y1, Z1, cmap=cm.coolwarm, linewidth =0, antialiased = False, vmin=vmin, vmax=vmax)
cset2 = ax.plot_surface(X2, Y2, Z2, cmap=cm.coolwarm, linewidth =0, antialiased = False, vmin=vmin, vmax=vmax)

所以您的完整代码将如下所示：

fig = plt.figure()
ax = fig.add_subplot(111, projection = '3d')
x1=np.arange(0,2,1)
x2 = np.arange(0,10,1)
y1 =np.arange(0,5,1)
y2=np.arange(5,12,1)
Y1, X1 = np.meshgrid(x1,y1)
Y2, X2 = np.meshgrid(x2,y2)
gridValue1 = gridValue[:10]
gridValue2 = gridValue[10:]
Z1 = gridValue1.reshape(X1.shape)
Z2 = gridValue2.reshape(X2.shape)
vmin = gridValue.min()
vmax = gridValue.max()
cset1 = ax.plot_surface(X1, Y1, Z1, cmap=cm.coolwarm, linewidth =0, antialiased = False, vmin=vmin, vmax=vmax)
cset2 = ax.plot_surface(X2, Y2, Z2, cmap=cm.coolwarm, linewidth =0, antialiased = False, vmin=vmin, vmax=vmax)
ax.clabel(cset1, fontsize=9, inline=1)
ax.clabel(cset2, fontsize=9, inline=1)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('σ ')
plt.show()