python - 如何让 matplotlib 热图子图的大小反射(reflect)每个子图数据中的行数？

Question

我在 python 中使用 matplotlib 为我使用 k-means 聚类创建的不同聚类创建热图。现在我可以生成这个数字:

但我希望每个簇中的行数反射(reflect)在热图的大小中，而不是将它们全部缩放到相同的大小。 GridSpec 是执行此操作的正确方法吗？这是我在尝试谷歌解决方案时能找到的唯一一件事，但它似乎更适合网格上有子图并且希望某个子图跨越网格上多个行或列的情况。在这种情况下，我将创建一个包含数千行的网格，并告诉每个子图跨越数百行。这仍然是最好的方法吗？

编辑:如果我的问题不清楚，我最终会尝试创建一个像这样的图形。请注意，在左图中很容易看出簇 E 大于簇 F:

Answer 1

GridSpec 有一个参数 height_ratios。您可以将其设置为热图垂直形状的列表。

import numpy as np
import matplotlib.pyplot as plt

data = [np.random.rand(n,8) for n in [3,7,10,4]]

fig, axes = plt.subplots(nrows=len(data), 
                         gridspec_kw=dict(height_ratios=[d.shape[0] for d in data]))

for ax, d in zip(axes, data):
    ax.imshow(d)
    ax.tick_params(labelbottom=False)

plt.show()