如何将Python中Seaborn热力图改为白（0）绿（1）双色配色？

要实现这种无过渡的双色配色，核心是自定义颜色映射并绑定数值边界，具体操作如下：

• 先确认数据格式：
  Seaborn的heatmap仅支持二维数组输入，如果你的df.values是一维结构，需要先转换为二维，比如单列数据可以用df.values.reshape(-1, 1)，多列数据则根据实际维度调整形状。

• 自定义双色映射与数值边界：
  借助matplotlib的工具类创建仅包含白色和绿色的色阶，再通过边界规则让0和1各自对应唯一颜色：

  import seaborn as sns
  import matplotlib.pyplot as plt
  from matplotlib.colors import ListedColormap, BoundaryNorm
  
  # 定义颜色：索引0对应白色，索引1对应绿色（也可以用十六进制色值自定义深浅）
  cmap = ListedColormap(['white', 'green'])
  # 设置边界区间，让0落在[-0.5, 0.5]，1落在[0.5, 1.5]，避免边界值混淆
  bounds = [-0.5, 0.5, 1.5]
  norm = BoundaryNorm(bounds, cmap.N)
  
  # 绘制热力图（以my_array为例，一维df.values需先转二维）
  ax = sns.heatmap(my_array, cmap=cmap, norm=norm, cbar=True)
  plt.show()
  

• 优化颜色条显示（可选）：
  默认颜色条的刻度可能不准确，手动调整为0和1的对应标签：

  # 获取颜色条对象
  cbar = ax.collections[0].colorbar
  cbar.set_ticks([0, 1])
  cbar.set_ticklabels(['0', '1'])
  

注意：如果数据中存在0和1之外的数值，需要先将这些值映射为0或1，否则会显示超出边界的颜色，不符合双色需求。

内容的提问来源于stack exchange，提问作者Economist_Ayahuasca