[英]Customized color palette in seaborn heatmap
可以使用LinearSegmentedColormap.from_list
。 为了使绿色恰好位于中心,需要将vmin
和vmax
设置为对称为零。
import seaborn as sns; sns.set()
import matplotlib.pyplot as plt
from matplotlib.colors import LinearSegmentedColormap
import numpy as np
corr = np.corrcoef(np.random.random(((5, 5))))
fig, ax = plt.subplots(figsize=(6,6))
cmap = LinearSegmentedColormap.from_list('RedGreenRed', ['crimson', 'lime', 'crimson'])
ax = sns.heatmap(corr, cmap=cmap, vmin=-1, vmax=1, annot=True, fmt="0.2f", linewidths=.5)
plt.show()
PS:在红色和绿色之间添加黄色会更好看: LinearSegmentedColormap.from_list('', ['crimson', 'gold', 'lime', 'gold', 'crimson'])
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.