我正在使用 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,
)
最佳答案
一个想法是绘制另一个热图,使用透明颜色,并且只有原始数据帧为 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'])
。
关于pandas - Seaborn:在热图上注释缺失值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/65169929/