如何在 seaborn.heatmap 中设置颜色映射?

How to set mapping to colors in seaborn.heatmap?

我正在尝试使用 sns.heatmap() 绘制相关矩阵。具体来说,我运行的代码是

corr = train.corr()
sns.heatmap(corr, cmap = "RdBu_r")

这张returns下图:

我想要的是 0 对应白色,正相关对应红色阴影,负相关对应蓝色阴影。我浏览了 seaborn 的文档,但是,我无法弄明白。

有没有办法在 seaborn 中设置这个?

尝试使用 vminvmax 属性,如前所述 here

更稳健的解决方案可能是使用 heatmap 中的 center 参数来设置绘制发散数据的中心。

A = np.random.normal(1, 3, [5,5])
sns.heatmap(A, center=0, cmap=sns.diverging_palette(220, 20, as_cmap=True))

结果 the following figure