You can use matplotlib's ListedColormap as follows:
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)
which yields:
You can replace the strings 'green', 'yellow', 'red'
with hexcolors such as '#FF0000'
(equivalent to 'red'
) or rgb colors such as (1.,0.,0.)
(also equivalent to 'red'
).
与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…