在seaborn热图中屏蔽注释

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

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)

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

请指教

我尝试了一些方法，包括在annot大批。不幸的是它们不起作用。

这可能是最简单的方法。它涉及抓住texts轴的，应该只是标签annot which sns.heatmap放在那里。

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, show_annot_array.ravel()):
    text.set_visible(show_annot)

plt.show()