Python - Seaborn:修改热图传奇

时间:2016-05-05 05:27:46

标签: python seaborn

我刚创建了以下热图。 enter image description here

在图例中,max(vmax)设置为0.10。我这样做是因为我想避免更多地着色"极端"值。但是在图例中,是否可以对其进行修改并编写"> = 0.10"所以添加"大于或等于"?

1 个答案:

答案 0 :(得分:6)

所以这是一个非常hacky的解决方案,并认为几乎可以肯定有一种更聪明的方法来做到这一点,希望@mwaskom可以权衡,但我能够通过在调用时明确地将其作为参数传递来访问颜色条对象热图功能如下:

import seaborn as sns; sns.set()
import numpy as np; np.random.seed(0)
from matplotlib import pyplot as plt

fig, ax = plt.subplots()
fig.set_size_inches(14, 7)
uniform_data = np.random.rand(10, 12)
cbar_ax = fig.add_axes([.92, .3, .02, .4])
sns.heatmap(uniform_data, ax=ax, cbar_ax=cbar_ax)

制作:

enter image description here

我能够在ax.get_yticks()中找到自己的标记:

In [41]: cbar_ax.get_yticks()
Out [41]: array([ 0.19823662,  0.39918933,  0.60014204,  0.80109475])

标签本身就是字符串:

In [44]: [x.get_text() for x in cbar_ax.get_yticklabels()]
Out [44]: [u'0.2', u'0.4', u'0.6', u'0.8']

所以我们可以简单地在yticklabels中更改最后一个元素的文本对象,并希望得到一个更正的轴,这是我的最终代码:

fig, ax = plt.subplots()
fig.set_size_inches(14, 7)
uniform_data = np.random.rand(10, 12)
#add an axis to our plot for our cbar, tweak the numbers there to play with the sizing. 
cbar_ax = fig.add_axes([.92, .3, .02, .4])
#assign the cbar to be in that axis using the cbar_ax kw
sns.heatmap(uniform_data, ax=ax, cbar_ax=cbar_ax)

#hacky solution to change the highest (last) yticklabel
changed_val = ">= " + cbar_ax.get_yticklabels()[-1].get_text()

#make a new list of labels with the changed value.
labels = [x.get_text() for x in cbar_ax.get_yticklabels()[:-1]] + [changed_val]

#set the yticklabels to the new labels we just created. 
cbar_ax.set_yticklabels(labels)

产生:

enter image description here

可以找到关于这个主题的一些额外资源here,其中我从mwaskom的回复中提取了一些信息。