python - 具有不断变化的 y 值的 Matplotlib 热图

Question

我正在尝试为从两个表面之间进行的测量绘制一些数据。系统中的 z 方向定义为垂直于表面。问题是，沿着我绘图的 x 轴，我改变了两个表面之间的分离距离，这意味着对于每个切片，y 轴的最小值/最大值都会发生变化。我通过提供一个标准化的 y 轴来规避这个问题，其中 z_min 是底面，z_max 是顶面:

但是，这种表示在某种程度上扭曲了数据。理想情况下，我想在 y 轴上显示到墙的实际距离，并将系统边界外的区域留白。我(粗略地)勾勒出了我在这里的设想(当然，热图上的实际分布看起来应该有所不同):

我可以很容易地绘制出我想要的 3D 散点图，如下所示:

但我如何将数据转化为热图的可绘制形式？

我猜我必须炸毁 MxN 数组并通过插值法填充缺失值或简单地将它们标记为 NAN？但是后来我也不太确定如何在我的配色方案中添加硬截止以使系统之外的所有内容都变成白色。

Answer 1

你可以用 pcolormesh 来做到这一点，它将四边形的角作为参数

X, Y = np.meshgrid(np.linspace(0, 10, 100), np.linspace(0, 2*np.pi, 150),)
h = np.sin(Y)
Y *= np.linspace(.5, 1, 100)

fig, ax = plt.subplots(1, 1)
ax.pcolormesh(X, Y, h)