如何移除前两个子图的colorbar并保持heatmap位置不变

嗨，这个问题很好解决！核心就是利用seaborn.heatmap的cbar参数——默认它是True（自动生成colorbar），我们只需要给前两个热力图设置cbar=False，不让它们生成colorbar，第三个保持默认（或者显式写cbar=True）就行，这样既不会影响热力图的位置，又能只保留最后一个的colorbar。

修改后的完整代码 #

import matplotlib.pyplot as plt
import seaborn as sns

# 假设你已经定义好df_norm、df_tgfup、df_tgfdown、col这些变量
fig = plt.figure()
grid = plt.GridSpec(4, 6)

# 第一个热力图：关闭colorbar
plt.subplot(grid[0:2,0:5])
ax1 = sns.heatmap(df_norm, cmap='Reds', vmin=0.05, vmax=0.7, cbar=False)
ax1.axes.get_yaxis().set_visible(False)
ax1.xaxis.tick_top()
ax1.set_xticklabels(col, rotation=90)

# 第二个热力图：关闭colorbar
plt.subplot(grid[2,0:5])
ax2 = sns.heatmap(df_tgfup, cmap='Reds', vmin=0.05, vmax=0.7, cbar=False)
ax2.axes.get_xaxis().set_visible(False)
ax2.axes.get_yaxis().set_visible(False)
ax2.xaxis.tick_top()
ax2.set_xticklabels(col, rotation=90)

# 第三个热力图：保留colorbar（默认就是True，也可以显式写出）
plt.subplot(grid[3,0:5])
ax3 = sns.heatmap(df_tgfdown, cmap='Reds', vmin=0.05, vmax=0.7, cbar=True)
ax3.axes.get_xaxis().set_visible(False)
ax3.axes.get_yaxis().set_visible(False)
ax3.xaxis.tick_top()
ax3.set_xticklabels(col, rotation=90)

plt.tight_layout()
plt.show()

为什么这样做有效？ #

• 当设置cbar=False时，seaborn不会为这个热力图创建colorbar，所以前两个图会完全占据你用GridSpec分配的grid[0:2,0:5]和grid[2,0:5]空间，不会留出colorbar的位置，完美维持你想要的布局。
• 第三个图的colorbar会自动放在GridSpec的右侧（也就是grid[:,5]的区域），刚好和你目标图的效果匹配。
• 最后加个plt.tight_layout()能让整体布局更协调，避免元素重叠。

效果对比 #

你当前生成的图：

你想要实现的图：

运行修改后的代码，就能得到和你目标图完全一致的结果啦！

内容的提问来源于stack exchange，提问作者Shoujun Gu