Seaborn 热图...白色的底行和右列

Question

我通过sns.heatmap 在 Seaborn 有一张热图。我现在想将底行和右列变白，但保留值。

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

np.random.seed(2021)
df = pd.DataFrame(np.random.normal(0, 1, (6, 4))
df = df.rename(columns = {0:"a", 1:"b", 2:"c", 3:"d"})
df.index = [value for key, value in {0:"a", 1:"b", 2:"c", 3:"d", 4:"e", 5:"f"}.items()]
sns.heatmat(df, annot = True)
plt.show

我以为我必须在我的sns.heatmap 调用中包含一个mask 参数，但我没有成功给出正确的mask，并且掩码删除了注释。我还需要保留我的数据框d 的文本索引。如何在保留文本索引的同时使这些单元格变白？

Answer 1

这是一种方法：

• 使用原始数据进行标注（annot=data）
• 使用原始数据创建"norm"，用于着色
• 创建颜色图的副本并将"over" color 指定为“白色”
• 创建数据的副本，并用高于原始数据最大值的值填充右列和下一行（np.inf不能使用，因为那样不会放置注释）；使用此副本进行着色； seaborn 会神奇地为注释使用适当的颜色
• 要在热图中使用数据框的列和索引名称，只需使用sns.heatmap(..., xticklabels=df.columns, yticklabels=df.index)
• 如果您没有安装最新的 seaborn 版本，您可以考虑使用matplotlib's standard colormaps 之一，或者通过matplotlib.colors.ListedColormap() 创建一个。也许cmap = ListedColormap(sns.color_palette('rocket', 256))？

在示例代码中：

import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd
import numpy as np
from copy import copy

np.random.seed(2021)
df = pd.DataFrame(np.random.normal(0, 1, (6, 4)), columns=[*"abcd"], index=[*"abcdef"])
data = df.to_numpy()
data_for_colors = data.copy()
data_for_colors[:, -1] = data.max() + 10
data_for_colors[-1, :] = data.max() + 10
norm = plt.Normalize(data[:-1, :-1].min(), data[:-1, :-1].max())
# cmap = sns.color_palette('rocket', as_cmap=True).copy()
cmap = copy(plt.get_cmap('RdYlGn'))
cmap.set_over('white')
sns.set_style('white')
sns.heatmap(data=data_for_colors, xticklabels=df.columns, yticklabels=df.index,
            annot=data, cmap=cmap, norm=norm)
plt.show()