代码之家  ›  专栏  ›  技术社区  ›  RadRuss

如何在seaborn中将贴图设置为颜色。热图?

  •  4
  • RadRuss  · 技术社区  · 6 年前

    我正试图用sns绘制一个相关矩阵。热图()。具体来说,我运行的代码是

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

    这将返回以下图像: correlation matrix

    我想要的是0代表白色,正相关代表红色,负相关代表蓝色。我查阅了seaborn的文档,但是我没能弄明白。

    有没有办法在seaborn设定这个?

    2 回复  |  直到 3 年前
        1
  •  4
  •   johnnyheineken    6 年前

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

        2
  •  2
  •   Axel Demborg    6 年前

    更可靠的解决方案可能是使用 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