从 seaborn.heatmap 中的图中删除遮罩项

Question

如果使用 mask 选项遮盖 seaborn 热图中的某些元素，则遮罩会在图的顶部绘制白色 space，如在此快速中所见例如。

# generate some random data
x = [np.random.rand() for x in range(0,20)]
y = [np.random.rand() for y in x]
data = pd.DataFrame([x,y]) # cast it into a dataframe
corr = data.corr() # get the correlation values


# generate a mask
mask = []
l = [True, True, True, True, True, True, True, True, True, True, True, True, True, True, True, True, True, True, False, False]
for i in range(0,10):
    mask.append([True for x in range(0,20)])
for i in range(0,10):
    mask.append(l)
mask = np.array(mask)

sb.heatmap(corr,mask=mask)

有没有办法绘制图像，以便从绘图中删除遮罩区域，使其看起来更像以下内容：

Answer 1

您可以从整个数据框中删除 mask 为 True 的所有列和行。

masked_corr = corr.loc[~np.all(mask, axis=1), ~np.all(mask, axis=0)]
sns.heatmap(masked_corr)

会产生

从 seaborn.heatmap 中的图中删除遮罩项

Remove masked items from the figure in seaborn.heatmap

python

mask

matplotlib

seaborn