Python：如何将颜色条居中于Seaborn Heatmap的定义值？

Question

我正在尝试使用seaborn绘制热图。所以，我有从0到5的值，我需要制作一个蓝色到红色的色标，白色为1，蓝色 - 低于1，红色 - 从1到5。 我怎么能这样做？

import seaborn as sns; sns.set()
hm=sns.heatmap(proportion, vmin=0, vmax=5, cmap='RdBu')

这是我尝试做的，但它不是定制的......＆＃39;比例＆＃39;是我的变量。我有什么可以做的吗？

Answer 1

也许你的意思是使用center参数，

import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np

data = np.random.rand(10,10)*5

ax = sns.heatmap(data, vmin=0,vmax=5,center=1,cmap="RdBu_r")

plt.show()