Matplotlib添加colorbar致子图变形及布局问题求助

问题1：Colorbar遮挡轴标签的处理 #

方案A：调整Colorbar位置与间距 #

无需额外添加子图，绘制colorbar时通过pad参数控制与主图的间距，或指定底部放置并缩小宽度，避免遮挡轴标签：

import matplotlib.pyplot as plt
from matplotlib.gridspec import GridSpec

fig = plt.figure(figsize=(10, 8))
# 2行2列布局，宽高比适配顶部/侧边柱状图与中心矩阵
gs = GridSpec(2, 2, width_ratios=[0.1, 0.9], height_ratios=[0.1, 0.9], hspace=0.05, wspace=0.05)
ax_top = fig.add_subplot(gs[0, 1])
ax_side = fig.add_subplot(gs[1, 0])
ax_main = fig.add_subplot(gs[1, 1])

# 绘制顶部柱状图、侧边柱状图、中心矩阵
# ... 你的绘图代码 ...

# 添加底部colorbar，调整pad避免遮挡x轴标签
im = ax_main.imshow(mat_store)
cbar = fig.colorbar(im, ax=ax_main, location='bottom', pad=0.15, shrink=0.8)

方案B：调整轴标签偏移量 #

若不想改动colorbar，直接给中心轴的x轴标签增加labelpad参数，让标签下移：

ax_main.set_xlabel('X轴标签', labelpad=30)  # 根据遮挡程度调整pad数值

问题2：添加第三行Gridspec后的子图对齐问题 #

通过GridSpec的height_ratios精确分配行高，前两行保持原布局的高度比例，第三行仅留极小空间给colorbar；同时强制顶部/侧边柱状图与中心矩阵的轴范围一致，解决对齐问题：

fig = plt.figure(figsize=(10, 9))
# 3行2列布局，前两行高度比例匹配原设计，第三行用于colorbar
gs = GridSpec(3, 2, width_ratios=[0.1, 0.9], height_ratios=[0.1, 0.8, 0.1], 
              hspace=0.05, wspace=0.05)

ax_top = fig.add_subplot(gs[0, 1])       # 顶部柱状图
ax_side = fig.add_subplot(gs[1, 0])      # 侧边柱状图
ax_main = fig.add_subplot(gs[1, 1])      # 中心矩阵
ax_cbar = fig.add_subplot(gs[2, :])      # colorbar子图（跨两列）

# 绘制顶部柱状图，隐藏x轴避免重复
ax_top.bar(range(mat_store.shape[1]), mat_store.sum(axis=0))
ax_top.set_xticks([])

# 绘制侧边柱状图，隐藏y轴避免重复
ax_side.barh(range(mat_store.shape[0]), mat_store.sum(axis=1))
ax_side.set_yticks([])

# 绘制中心矩阵
im = ax_main.imshow(mat_store)

# 在colorbar子图中添加水平colorbar
cbar = fig.colorbar(im, cax=ax_cbar, orientation='horizontal')
# 移除colorbar子图边框，优化视觉效果
for spine in ax_cbar.spines.values():
    spine.set_visible(False)

# 强制对齐轴范围，确保顶部柱状图与中心矩阵x轴匹配
ax_top.set_xlim(ax_main.get_xlim())
# 强制侧边柱状图与中心矩阵y轴匹配
ax_side.set_ylim(ax_main.get_ylim())

这样设置后，第三行不会过度压缩中心轴空间，同时通过轴范围对齐解决了子图错位问题。

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