將值映射到 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' ）等十六進制顏色或(1.,0.,0.)等 rgb 顏色（也相當於'red' ）。