python - 将值映射到 seaborn 热图中的特定颜色

Question

我正在使用 seaborn 包在 Python 中绘制热图。我绘制的值是离散的，它们是整数 -1、0 和 1。

我希望热图中值为 -1 的单元格显示为绿色，0 为黄色，1作为红色。

是否可以在 cubehelix_palette() 或 colour_palette() 函数中指定此规则？

Answer 1

您可以使用 matplotlib.colors.ListedColormap如下:

import numpy as np
import seaborn as sns
from matplotlib.colors import ListedColormap

data = np.random.randint(-1, 2, (10,10)) # Random [-1, 0, 1] data
sns.heatmap(data, cmap=ListedColormap(['green', 'yellow', 'red']), annot=True)

产生:

您可以将字符串 'green', 'yellow', 'red' 替换为十六进制颜色，例如 '#FF0000'(相当于 'red' ) 或 rgb 颜色，例如 (1.,0.,0.)(也相当于 'red')。