pandas - Seaborn:在热图上注释缺失值

Question

我正在使用 seaborn 库在 python 中绘制热图。数据框包含一些缺失值 (NaN)。我希望与这些字段对应的热图单元格是白色的(默认情况下)并且还用字符串 NA 进行注释。但是，如果我看对了，注释不适用于缺失值。有什么破解方法吗？

我的代码:

    sns.heatmap(
        df,
        ax=ax[0, 0],
        cbar=False,
        annot=annot_df,
        fmt="",
        annot_kws={"size": annot_size, "va": "center_baseline"},
        cmap="coolwarm",
        linewidth=0.5,
        linecolor="black",
        vmin=-max_value,
        vmax=max_value,
        xticklabels=True,
        yticklabels=True,
    )

Answer 1

一个想法是绘制另一个热图，使用透明颜色，并且只有原始数据帧为 NaN 的值。要控制轴标签，应最后绘制“真实”热图。请注意，NaN 单元格的颜色是绘图的背景色。

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

data = np.where(np.random.rand(7, 10) < 0.2, np.nan, np.random.rand(7, 10) * 2 - 1)
df = pd.DataFrame(data)
annot_df = df.applymap(lambda f: f'{f:.1f}')
fig, ax = plt.subplots(squeeze=False)
sns.heatmap(
    np.where(df.isna(), 0, np.nan),
    ax=ax[0, 0],
    cbar=False,
    annot=np.full_like(df, "NA", dtype=object),
    fmt="",
    annot_kws={"size": 10, "va": "center_baseline", "color": "black"},
    cmap=ListedColormap(['none']),
    linewidth=0)
sns.heatmap(
    df,
    ax=ax[0, 0],
    cbar=False,
    annot=annot_df,
    fmt="",
    annot_kws={"size": 10, "va": "center_baseline"},
    cmap="coolwarm",
    linewidth=0.5,
    linecolor="black",
    vmin=-1,
    vmax=1,
    xticklabels=True,
    yticklabels=True)
plt.show()

PS:要明确地为 'NA' 单元格着色，例如可以使用 cmap=ListedColormap(['yellow'])。