遮盖seaborn热图中的注释

Question

我想制作一个仅在特定单元格中具有注释的热图。 尽管执行此操作的一种方法是在所有单元格中制作带有注释的热图，然后覆盖另一个没有注释但在希望原始注释可见的区域中被屏蔽的热图：

import numpy as np
import seaborn as sns

par_corr_p = np.array([[1, 2], [3, 4]])
masked_array = np.ma.array(par_corr_p, mask=par_corr_p<2)

fig, ax = plt.subplots()
sns.heatmap(par_corr_p, ax=ax, cmap ='RdBu_r', annot = par_corr_p, center=0, vmin=-5, vmax=5)
sns.heatmap(par_corr_p, mask = masked_array.mask, ax=ax,  cmap ='RdBu_r', center=0, vmin=-5, vmax=5)

但是，这不起作用-第二个热图没有覆盖第一个热图：

请指教

Answer 1

我尝试了一些方法，包括在annot数组中使用numpy.nan或“”。 不幸的是，它们不起作用。

这可能是最简单的方法。 它涉及到抓取轴的texts ，这些texts只能是sns.heatmap放在其中的annot中的标签。

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

par_corr_p = np.array([[1, 2], [3, 4]])

data = par_corr_p
show_annot_array = data >= 2

fig, ax = plt.subplots()
sns.heatmap(
    ax=ax,
    data=data,
    annot=data,
    cmap ='RdBu_r', center=0, vmin=-5, vmax=5
)
for text, show_annot in zip(ax.texts, (element for row in show_annot_array for element in row)):
    text.set_visible(show_annot)

plt.show()