海生热图中特定颜色的地图值

Question

我正在用Python和seaborn包绘制热图。我的价值观
绘图是离散的，它们是整数“-1”、“0”和“1”。
我希望热图中的单元格显示为绿色，值为“-1”，
“0”表示黄色，“1”表示红色。
是否可以在“cubehelix\u palete（）”中指定此规则，或者
colour\u palete（）函数？

Answer 1

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:

您可以将字符串“green”、“yellow”、“red”替换为诸如“#FF0000”（相当于“red”）或rgb颜色，如（1,0,0.）`（也可以是相当于“红色”。