如何为Seaborn Heatmap添加独立的额外行与列？

Seaborn本身没有专门为这种“独立额外行列”提供直接API，但可以通过Matplotlib轴对象结合Seaborn绘图逻辑实现更规范的方案，比手动添加行列加分隔线更优雅：

方案1：子图拼接（推荐） #

把额外行列作为独立子图和主heatmap拼接，保持视觉连贯：

• 用matplotlib.gridspec.GridSpec划分布局，分配主图与额外行列的空间占比。
• 分别在对应子图绘制主heatmap、额外行/列数据，可搭配seaborn.heatmap或Matplotlib原生绘图函数。
• 调整子图间距，隐藏多余坐标轴刻度，让整体呈现统一视觉效果。

示例代码：

import seaborn as sns
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
import pandas as pd
import numpy as np

# 模拟数据
main_data = pd.DataFrame(np.random.rand(5,5))
extra_row = pd.DataFrame(np.random.rand(1,5), columns=main_data.columns)
extra_col = pd.DataFrame(np.random.rand(5,1), index=main_data.index)

# 划分布局：主图(0,0)、额外行(1,0)、额外列(0,1)
gs = gridspec.GridSpec(2, 2, width_ratios=[5,1], height_ratios=[5,1], wspace=0.05, hspace=0.05)

# 绘制主heatmap
ax_main = plt.subplot(gs[0,0])
sns.heatmap(main_data, ax=ax_main, cbar=False, annot=True)

# 绘制额外行
ax_row = plt.subplot(gs[1,0])
sns.heatmap(extra_row, ax=ax_row, cbar=False, annot=True, xticklabels=False)
ax_row.set_yticklabels(["额外行"], rotation=0)

# 绘制额外列
ax_col = plt.subplot(gs[0,1])
sns.heatmap(extra_col, ax=ax_col, cbar=False, annot=True, yticklabels=False)
ax_col.set_xticklabels(["额外列"], rotation=0)

# 统一调整样式
ax_main.tick_params(labelbottom=False)
ax_row.tick_params(labelleft=False)
plt.show()

方案2：利用mask参数（简单场景适配） #

若额外行列仅需与主区域区分，无需独立配色/样式，可在原DataFrame添加行列后，用mask参数区分区域样式：

• 将额外行列数据并入原DataFrame。
• 创建同尺寸的mask矩阵，标记主区域与额外区域。
• 绘制heatmap时，通过mask、cmap等参数分别设置不同区域样式。

示例代码：

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

# 主数据合并额外行列
main_data = pd.DataFrame(np.random.rand(5,5))
main_data.loc["额外行"] = np.random.rand(5)
main_data["额外列"] = np.random.rand(6)

# 创建mask：主区域设为False，额外行列设为True
mask = np.zeros_like(main_data, dtype=bool)
mask[:5, :5] = True

# 分区域绘制heatmap
sns.heatmap(main_data, mask=~mask, cmap="Blues", annot=True)
sns.heatmap(main_data, mask=mask, cmap="Greys", annot=True, cbar=False, alpha=0.7)
plt.show()

以上两种方案均基于Seaborn与Matplotlib原生功能实现，符合官方绘图逻辑，比手动添加分隔线更易维护和调整样式。

内容的提问来源于stack exchange，提问作者walves