如何调整seaborn热图中各个子图的高度

Question

我有一个使用 seaborn 的热图，我正在尝试调整下面第四个图的高度。您会看到它只有 2 行数据，而其他数据行更多：

我使用以下代码来创建情节：

f, ax = plt.subplots(nrows=4,figsize=(20,10))
cmap = plt.cm.GnBu_r
sns.heatmap(df,cbar=False,cmap=cmap,ax=ax[0])
sns.heatmap(df2,cbar=False,cmap=cmap,ax=ax[1])
sns.heatmap(df3,cbar=False,cmap=cmap,ax=ax[2])
sns.heatmap(df4,cbar=False,cmap=cmap,ax=ax[3])

有谁知道下一步基本上使第 4 个地块的高度更小，从而伸展其他 3 个地块？第 4 个地块通常总是有 2-3 个，而其他地块通常有 6-7 个。非常感谢！

Answer 1

像往常一样，matplotlib 非常时髦/乏味。但是就在这里！

f = plt.figure(constrained_layout = True)  
specs = f.add_gridspec(ncols = 1, nrows = 4, height_ratios = [1,1,1,.5])

for spec, df in zip(specs, (df, df2, df3, df4)):
  ax = sns.heatmap(df,cbar=False,cmap=cmap, ax=f.add_subplot(spec)) 

您可以使用 height_ratio 更改相对于彼此的高度。如果您想更改相对宽度，还可以实现 wdith_ratios 参数。您还可以实现一个 for 循环来遍历图形。