如何控制Seaborn热力图行高度？按底层计数行总和调整

我来帮你搞定这个需求！Seaborn的heatmap本身没有直接设置行高度的参数，但我们可以借助Matplotlib的底层API手动调整行高比例，让每行高度对应原始数据的行总和，以此体现各行的重要性。下面是完整的实现步骤和代码：

步骤说明 #

1. 计算行百分比与行总和：先得到你需要的行百分比热力图数据，同时保留原始数据的行总和（用来决定行高）。
2. 归一化行总和：将行总和转化为相对比例（总和为1），这样可以直接作为行高的权重。
3. 生成热力图并调整行高：通过修改Matplotlib的gridspec参数设置行高比例，同时调整y轴刻度位置确保标签对齐。

完整代码示例 #

import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd

# 替换成你自己的数据集
data = pd.DataFrame({
    'Category1': [20, 45, 10],
    'Category2': [30, 15, 35],
    'Category3': [10, 20, 15]
}, index=['GroupA', 'GroupB', 'GroupC'])

# 计算行百分比（你的原有逻辑）
pcts = data.apply(lambda x: x / float(x.sum()), axis=1)
# 提取每行的原始计数总和
row_totals = data.sum(axis=1)

# 归一化行总和，得到行高的相对比例（总和为1）
row_heights = row_totals / row_totals.sum()

# 创建画布，可根据数据量调整figsize
fig, ax = plt.subplots(figsize=(8, 6))

# 绘制行百分比热力图（注意传入pcts而非原始data）
sns.heatmap(pcts, ax=ax, cmap='YlOrRd', annot=True, fmt='.1%')

# 核心操作：修改gridspec的height_ratios参数调整行高
ax.get_gridspec().update(height_ratios=row_heights)

# 调整y轴刻度位置，让标签对齐每行中心
yticks_positions = row_heights.cumsum() - row_heights / 2
ax.set_yticks(yticks_positions)
ax.set_yticklabels(data.index)

# 可选：优化标签显示，避免重叠
ax.tick_params(axis='y', labelsize=10)

plt.tight_layout()
plt.show()

关键细节解释 #

• 归一化行总和的原因：Matplotlib的gridspec的height_ratios接受相对比例值，归一化后能确保所有行的高度比例总和为1，完美适配画布整体高度。
• 调整y轴刻度的必要性：行高度改变后，默认的y轴刻度位置会偏移，重新计算位置可以让行标签准确对齐每行中心。
• 自定义空间：你可以根据需求修改cmap（颜色映射）、annot（是否显示百分比文本）、fmt（文本格式）以及figsize（画布大小）等参数。

内容的提问来源于stack exchange，提问作者Robin Tschötschel