颜色映射整个子图

Question

我在使用彩色地图时遇到了一些问题。基本上，我想要制作的内容类似于下图。

在底部子图上，我希望能够绘制相关颜色，但跨越子图的整个背景。即它看起来就像整个图上的颜色图，没有绘制线或点。它仍应与散点图中显示的颜色相对应。

可以这样做吗？我最理想的做法是把这个背景放在最上面的子图下。 （y 比例单位不同）

感谢您的帮助。

底部散点图的

代码：

x = np.arange(len(wind))
y = wind
t = y
plt.scatter(x, y, c=t)

其中风是一维数组

Answer 1

您可以使用 imshow 来显示您的风阵。它需要重新整形为二维数组，但“高度”维度可以是长度 1。将范围设置为顶部轴的维度使其与其对齐。

wind = np.random.randn(100) + np.random.randn(100).cumsum() * 0.5
x = np.arange(len(wind))
y = wind
t = y

fig, ax = plt.subplots(2,1,figsize=(10,6))

ax[0].plot(x,y)

ax[1].plot(x, 100- y * 10, lw=2, c='black')

ymin, ymax = ax[1].get_ybound()
xmin, xmax = ax[1].get_xbound()

im = ax[1].imshow(y.reshape(1, y.size), extent=[xmin,xmax,ymin,ymax], interpolation='none', alpha=.5, cmap=plt.cm.RdYlGn_r)
ax[1].set_aspect(ax[0].get_aspect())

cax = fig.add_axes([.95,0.3,0.01,0.4])
cb = plt.colorbar(im, cax=cax)
cb.set_label('Y parameter [-]')

如果您想将其用作“背景”，您应该首先绘制您想要的任何内容。然后抓取底部绘图的范围并将其设置为imshow 的范围。您还可以使用cmap= 向imshow 提供您想要的任何颜色图。