如何在不同的子图中绘制 pcolor 颜色条 - matplotlib

我正在尝试将我的图拆分为不同的子图。我想要实现的是将子图的颜色条放在不同的子图中。 现在我正在使用：

# first graph
axes = plt.subplot2grid((4, 2), (0, 0), rowspan=3)
pc = plt.pcolor(df1, cmap='jet')

# second graph
axes = plt.subplot2grid((4, 2), (3, 0))
plt.pcolor(df2, cmap='Greys')

# colorbar
plt.subplot2grid((4, 2), (0, 1), rowspan=3)
plt.colorbar(pc) 

But the result is the following (notice the unwanted empty graph left to the colorbar):

如何只打印颜色条而不打印左图？

Thanks

colorbar()接受一个cax关键字参数允许您指定axes将在其上绘制颜色条的对象。

在您的情况下，您可以将 colorbar 调用更改为以下内容：

# colorbar
axes = plt.subplot2grid((4, 2), (0, 1), rowspan=3)
plt.colorbar(pc, cax=axes)

这将占据由下式给出的整个空间subplot2grid;您可以通过让主轴比颜色条轴占用更多的列，或者通过设置gridspec明确地。例如，您的图形可能更容易通过以下方式进行调整：

from matplotlib import gridspec
gs = gridspec.GridSpec(2, 2, height_ratios=(3, 1), width_ratios=(9, 1))

# first graph
axes = plt.subplot(gs[0,0])
pc = plt.pcolor(df1, cmap='jet')

# second graph
axes = plt.subplot(gs[1,0])
plt.pcolor(df2, cmap='Greys')

# colorbar
axes = plt.subplot(gs[0,1])
plt.colorbar(pc, cax=axes)

然后你就可以改变height_ratios and width_ratios根据您的喜好。