在 seaborn 热力图上标注缺失值

Question

在 seaborn 热力图上标注缺失值

3

我正在使用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,
    )

- maciek

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- JohanC · Accepted Answer

一个想法是绘制另一个热图，使用透明颜色，并仅使用原始数据帧中值为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'])。