Python 如何在热图右侧添加自定义刻度

发布于10月01日

给出以下示例数据

data =\
{'q1': [6, 4, 4, 4, 6, 6, 6, 4, 6, 6, 6, 6],
 'q2': [3, 3, 3, 4, 3, 3, 4, 3, 4, 3, 4, 1],
 'q3': [6, 3, 4, 4, 4, 4, 6, 6, 6, 6, 4, 1],
 'q4': [3, 6, 6, 6, 6, 6, 4, 4, 6, 4, 6, 4],
 'q5': [4, 3, 3, 3, 3, 6, 6, 4, 4, 6, 3, 4],
 'q6': [3, 3, 4, 4, 6, 3, 3, 1, 6, 4, 4, 4],
 'q7': [4, 3, 3, 3, 3, 6, 4, 1, 4, 1, 4, 1],
 'q8': [4, 4, 1, 6, 4, 6, 4, 1, 6, 1, 4, 1],
 'q9': [4, 6, 4, 3, 4, 3, 4, 6, 6, 4, 3, 4],
 'q10': [3, 4, 1, 3, 4, 3, 3, 6, 6, 3, 4, 4]}

以下是我的代码:

import seaborn as sns
import pandas as pd

df = pd.DataFrame(data)

cm = sns.heatmap(df.T, annot=False, cbar=True)

cm.get_figure().savefig('heatmap.pdf')

下面是输出结果:

我要做的是在右侧添加自定义y轴刻度，如下所示:

我怎么能这样做呢？我按照建议的here使用了tick_params，但这只使用了左侧相同的y-tick.

推荐答案

您可以创建孪生轴并设置其刻度和标签:

cm = sns.heatmap(df.T, annot=False, cbar=True)

# create the twin axis
ax_twin = cm.twinx()

# Set the limits of the y twin axis as the ones of the original y axis
ax_twin.set_ylim(cm.get_ylim())

# Set the position of the ticks in the twin axis 
# copying the positions of the ticks on the original axis
ax_twin.set_yticks(cm.get_yticks())  

# Set the labels of the just created ticks
ax_twin.set_yticklabels('abcdefghij')

# Save the figure, as before
cm.get_figure().savefig('heatmap.pdf')

这就是你所得到的: