如何对热图的不同行使用不同的色彩映射表

沃梦达教程 Python问题

2022-01-01 4

How to use a different colormap for different rows of a heatmap(如何对热图的不同行使用不同的色彩映射表)

本文介绍了如何对热图的不同行使用不同的色彩映射表的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！

问题描述

我正在尝试将热图中的1行更改为其他颜色

以下是数据集：

m = np.array([[ 0.7,  1.4,  0.2,  1.5,  1.7,  1.2,  1.5,  2.5],
              [ 1.1,  2.5,  0.4,  1.7,  2. ,  2.4,  2. ,  3.2],
              [ 0.9,  4.4,  0.7,  2.3,  1.6,  2.3,  2.6,  3.3],
              [ 0.8,  2.1,  0.2,  1.8,  2.3,  1.9,  2. ,  2.9],
              [ 0.9,  1.3,  0.8,  2.2,  1.8,  2.2,  1.7,  2.8],
              [ 0.7,  0.9,  0.4,  1.8,  1.4,  2.1,  1.7,  2.9],
              [ 1.2,  0.9,  0.4,  2.1,  1.3,  1.2,  1.9,  2.4],
              [ 6.3, 13.5,  3.1, 13.4, 12.1, 13.3, 13.4, 20. ]])
data = pd.DataFrame(data = m)

我现在使用的是海运热图，我只能创建如下内容：

cmap = sns.diverging_palette(240, 10, as_cmap = True)
sns.heatmap(data, annot = True, cmap = "Reds")
plt.show

我希望更改最后一行的配色方案，以下是我想要实现的(我在Excel中完成了此操作)：

我是否可以使用海运热图在Python中实现这一点？谢谢！

推荐答案

您可以一分为二，屏蔽不需要的部分，然后分别打印：

# Reds
data1 = data.copy()
data1.loc[7] = float('nan')
ax = sns.heatmap(data1, annot=True, cmap="Reds")

# Greens
data2 = data.copy()
data2.loc[:6] = float('nan')
sns.heatmap(data2, annot=True, cmap="Greens")