如何为seaborn heatmap中的蒙版单元分配自定义颜色?

时间:2018-09-18 19:04:08

标签: python seaborn heatmap

我有一个数据集,值是-4到4,还有一些nan值。我使用seaborn热图绘制热图。我需要使用的色图是从红色到白色再到蓝色。我的问题是被遮罩的单元格也是白色/灰色,很难区分,然后在颜色图中接近0的值。

有没有办法在不绘制热图两次的情况下将nan值分配为黑色?

1 个答案:

答案 0 :(得分:2)

您有两个选择。

  1. 使用颜色图的bad值。即如果将蒙版值设置为nan,它们将通过

    以设置为颜色图的颜色显示
    colormap.set_bad("black") 
    
  2. 将轴的背景设置为黑色,以使被掩蔽(因此未绘制)的值显示为透明,而背景色则透明,

    ax.set_facecolor("black")