如何在Seaborn热力图中隐藏NaN标注同时保留颜色编码？

要实现只隐藏NaN标注、保留对应单元格颜色的效果，核心是分开控制热力图的显示和标注的显示——不要把NaN的mask传给heatmap()的mask参数（这会同时隐藏颜色），而是把mask传给标注的配置参数，只控制标注是否显示。

具体实现步骤 #

1. 生成标记NaN位置的布尔数组：用np.isnan()识别annot数组（或数据数组）中的NaN位置
2. 绘制热力图时，不设置heatmap()的mask参数，确保所有单元格保留颜色
3. 在annot_kws中传入mask参数，指定哪些位置不显示标注

代码示例 #

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

# 构造带NaN的示例数据和annot数组
data = np.array([[1.2, 2.5, np.nan], [4.1, np.nan, 6.3], [7.8, 8.2, 9.5]])
annot_data = data.copy()  # 如果annot是独立数组，替换成你的annot即可

# 创建仅标记NaN位置的mask
nan_mask = np.isnan(annot_data)

# 绘制热力图
plt.figure(figsize=(8, 6))
sns.heatmap(
    data,
    annot=True,
    annot_kws={'mask': nan_mask},  # 仅隐藏NaN对应的标注
    cmap='coolwarm',
    fmt='.1f',  # 根据你的数据调整标注格式
    cbar=True
)
plt.title("热力图：隐藏NaN标注但保留单元格颜色")
plt.show()

关键说明 #

• 如果你直接把nan_mask传给heatmap()的mask参数，会让对应单元格的颜色和标注一起消失，这是你之前遇到的问题
• 通过annot_kws={'mask': nan_mask}，只对标注层做隐藏，热力图的颜色渲染不受影响，完美实现需求

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