python - seaborn 热图颜色图

Question

我有一个数据框 df值从 0 到 x(x 是整数，没有固定值)，在我的例子中是 x=10
我想用 cmap 'Reds' 映射热图，但是值 0 不应该是白色而是绿色 '#009933'

import seaborn as sns # matplotlib inline 
import random
data = []
for i in range(10):
    data.append([random.randrange(0, 11, 1) for _ in range(10)])
df = pd.DataFrame(data)

fig, ax = plt.subplots(figsize = (12, 10)) 
# cmap = [????]
ax = sns.heatmap(df, cmap='Reds', linewidths = 0.005, annot = True, cbar=True) 
                            
plt.show()

我怎样才能做到这一点？

Answer 1

作为 accepted answer 的替代品您也可以设置 vmin略高于0并使用 set_under 定义超出范围值的颜色:

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

cmap = copy.copy(plt.get_cmap("Reds"))
cmap.set_under('#009933')
sns.heatmap(np.random.randint(0,10,(10,10)), cmap=cmap, lw=0.005, annot=True, vmin=1e-5)

under -color 默认情况下不显示在颜色栏中。要更改颜色条，例如使用 sns.heatmap(..., cbar_kws={'extend':'min', 'extendrect':True}) .这些参数的解释可以在 colorbar docs 中找到。 .