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

Question

我试图用sns.heatmap（）绘制相关矩阵。具体来说，我运行的代码是

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

返回以下图片：

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

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

Answer 1

尝试使用vmin和vmax属性，如上所述here

Answer 2

一种更可靠的解决方案可能是使用center中的heatmap参数来设置绘制差异数据的中心。

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