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')

这是我尝试做的，但不是自定义的……“比例”是我的变量。有什么我可以做的吗？

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()