使单元格在 seaborn 热图之间具有相同的大小

Question

我正在制作两个 seaborn 热图，我将它们放入两个 matplotlib 子图中，如下所示：

fig, axes = plt.subplots(2, 1, sharex=True, figsize=(5,10))


data = [
    [0,0,0],
    [0,0,0],
    [0,0,0],
    [0,0,0],
    [0,0,0],
    [0,0,0],
    [0,0,0],
]
sns.heatmap(
        data, 
        ax=axes[0],
        linewidths=0.2,
        cbar=False)


data = [
    [0,0,0],
    [0,0,0],
    [0,0,0],
]
sns.heatmap(
        data, 
        ax=axes[1],
        linewidths=0.2,
        cbar=False)

axes[0].set_title("A")
axes[1].set_title('B')

如您所见，这会导致子图 A 和 B。子图 A 中的各个单元格的大小都相同。子图 B 中的单元格也都相同大小。但是，A 和 B 之间的单元格大小不同。如何确保子图之间的单元格大小相同？

我理解这意味着 A 地块总体上会比 B 地块大，这不是问题。

Answer 1

一种可能的解决方案是改变两个子图的大小比例。

如果图 A 和 B 之间的大小比率设置为与图 A 和 B 的行数之间的比率相同，则大小应该（或多或少）相同。

data_a = [
    [0,0,0],
    [0,0,0],
    [0,0,0],
    [0,0,0],
    [0,0,0],
    [0,0,0],
    [0,0,0],
]
data_b = [
    [0,0,0],
    [0,0,0],
    [0,0,0],
]
fig, axes = plt.subplots(2, 1, 
        sharex=True, 
        figsize=(5,10), 
        gridspec_kw={'height_ratios': [len(data_a), len(data_b)]})


sns.heatmap(
        data_a, 
        ax=axes[0],
        linewidths=0.2,
        cbar=False)

sns.heatmap(
        data_b, 
        ax=axes[1],
        linewidths=0.2,
        cbar=False)

axes[0].set_title("A")
axes[1].set_title('B')

我想知道是否还有一个可以切换以获取此行为的 seaborn 选项？