在 sns.heatmap 子图中设置 xtick 标签

Question

我正在尝试将sns.heatmap的 3 个子图与自定义xticklabels和yticklabels 。 您可以看到我输入的标签与一些默认标签重叠。 如何避免这种情况？

glue = sns.load_dataset("glue").pivot("Model", "Task", "Score")
xlabels=['a','b','c','d','e','f','g','h']
ylabels=['AA','BB','CC','DD','EE','FF','GG','HH']

fig, (ax1,ax2,ax3) = plt.subplots(nrows=1, ncols=3,figsize=(24,8))
ax1 = fig.add_subplot(1,3,1)
# figure(figsize=(6, 4), dpi=80)
ax1=sns.heatmap(glue, annot=True,cmap='Reds', cbar=False, linewidths=0.2, linecolor='white',xticklabels=xlabels, yticklabels=ylabels)

xlabels=['p','q','r','s','t','u','v','z']
ylabels=['PP','QQ','RR','SS','TT','UU','VV','ZZ']
ax2 = fig.add_subplot(1,3,2)
ax2=sns.heatmap(glue, annot=True,cmap='Blues', cbar=False, linewidths=0.2, linecolor='white',xticklabels=xlabels, yticklabels=ylabels)

xlabels=['10','20','30','40','50','60','70','80']
ylabels=['11','21','31','41','51','61','71','81']
ax3 = fig.add_subplot(1,3,3)
ax3=sns.heatmap(glue, annot=True,cmap='Greens', cbar=False, linewidths=0.2, linecolor='white',xticklabels=xlabels, yticklabels=ylabels)

Answer 1

无需add_subplot ，因为您已经使用plt.subplots创建了子图。 使用heatmap的ax参数：

fig, (ax1,ax2,ax3) = plt.subplots(nrows=1, ncols=3, figsize=(24,8))

kwargs = {
    'cbar': False,
    'linewidths': 0.2,
    'linecolor': 'white',
    'annot': True, 
}

xlabels=['a','b','c','d','e','f','g','h']
ylabels=['AA','BB','CC','DD','EE','FF','GG','HH']

sns.heatmap(glue, cmap='Reds', xticklabels=xlabels, 
            yticklabels=ylabels, ax=ax1, **kwargs)

xlabels=['p','q','r','s','t','u','v','z']
ylabels=['PP','QQ','RR','SS','TT','UU','VV','ZZ']

sns.heatmap(glue, cmap='Blues', xticklabels=xlabels, 
            yticklabels=ylabels, ax=ax2, **kwargs)

xlabels=['10','20','30','40','50','60','70','80']
ylabels=['11','21','31','41','51','61','71','81']

sns.heatmap(glue, cmap='Greens', xticklabels=xlabels, 
            yticklabels=ylabels, ax=ax3, **kwargs)

Output：