根据条件为seaborn热图更改颜色

Question

我已经使用以下代码使用seaborn生成了一个热力图：

sns.heatmap(df.sort_index(axis=1), cmap="YlOrRd_r", center=0.8, square=True, annot=annot_df.sort_index(axis=1), annot_kws={"size":22, "va": "center_baseline", "color":"white"}, fmt="", xticklabels=True, yticklabels=True, linewidth=1, linecolor="grey", vmax=1, vmin=0.5)

现在我想要以不同的颜色进行着色，例如将所有值 > 0.9 的单元格涂成蓝色，而其他单元格应保持红黄渐变。是否有一种简单的实现方法？提前感谢您！

- Simone F

Answer 1

您可以使用蓝色调色板再次绘制热图，省略注释并设置掩码只绘制值大于0.9的单元格（掩码隐藏不需要的单元格）。

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

labels = list('abcdef')
N = len(labels)
heatm = np.random.uniform(0.5, 1, (N, N))
ax = sns.heatmap(heatm, cmap="YlOrRd_r", center=0.8, square=True, annot=True,
                 annot_kws={"size": 12, "va": "center_baseline", "color": "white"}, fmt=".2f", 
                 xticklabels=labels, yticklabels=labels, linewidth=1, linecolor="grey", vmin=0.5, vmax=1, cbar=False)
ax = sns.heatmap(heatm, mask=heatm < 0.9, cmap='Blues', square=True, annot=False, vmin=0, vmax=1, cbar=False, ax=ax)
plt.show()

- JohanC