python中的两种颜色热图答案

【问题标题】：Two colour heat map in pythonpython中的两种颜色热图
【发布时间】：2022-12-16 20:11:01
【问题描述】：

我有以下数据：

my_array = array([[0, 0, 1, 0, 0],
       [0, 1, 1, 1, 0],
       [0, 0, 0, 1, 1],
       [0, 0, 1, 1, 1],
       [0, 1, 1, 0, 0],
       [1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1],
       [0, 0, 0, 0, 1],
       [0, 1, 0, 1, 0]])

和

df.values = array([246360,  76663,  29045,  11712,   5526,   3930,   3754,   1677,
         1328])

我正在制作这样的热图：

import seaborn as sns
import matplotlib.pyplot as plt
cmap = sns.cm.rocket_r
ax = sns.heatmap(my_array, xticklabels=["A", "B", "C", "D", "E"], yticklabels=df.values, cmap = cmap)
ax.set(xlabel='Test Type', ylabel='Number', title='patterns of missingness')
fig=plt.figure(figsize=(40,30), dpi= 20, facecolor='w', edgecolor='k')
fig

我得到以下信息：

我的问题是，如何摆脱连续的色标并只选择两种不同的颜色：0 为白色，1 为绿色？

【问题讨论】：

你能澄清一下 array 在这里的意思吗？另外，您能否确保该图可以使用问题中提供的代码重现？

标签： python heatmap

【解决方案1】：

您可以使用 LinearSegmentedColormap 而不是您正在使用的当前 cmap 来执行此操作。

import matplotlib as mpl
cmap = mpl.colors.LinearSegmentedColormap.from_list('my_cmap', ['white', 'green'], 2)

然后在调用sns.heatmap时将ticks参数传递给cbar_kws参数：

ax = sns.heatmap(my_array, xticklabels=["A", "B", "C", "D", "E"],
                 yticklabels=df.values, cmap=cmap, cbar_kws={"ticks":[0,1]})

这使：

如果很清楚绿色和白色代表什么，您也可以使用cbar=False 完全关闭 cbar。

ax = sns.heatmap(my_array, xticklabels=["A", "B", "C", "D", "E"],
             yticklabels=df.values, cmap=cmap, cbar=False)

【讨论】：