如何将颜色条居中于 seaborn 热图的定义值？

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