Matplotlib不修改第三方函数源码调整colorbar与坐标轴间距的方法

可以通过返回的图形句柄直接调整colorbar间距，不需要修改原模块的源代码，以下是两种常用实现方案：

方案1：调整全局子图布局（最简便） #

直接通过Figure对象的subplots_adjust方法缩小绘图区域的右边界，给右侧的colorbar预留足够空间，数值可以根据自己的实际显示效果调整：

import module1 as m1
import matplotlib as mpl

# 注意原示例中的figure.fisize是笔误，正确参数名为figure.figsize
with mpl.rc_context({'figure.figsize': (10,7), 'legend.borderaxespad': 4}):
    fig = m1.func_calc_and_draw(my_data)
# 调整右边界：数值范围0-1，数值越小右侧预留空间越大，根据实际显示效果调整
fig.subplots_adjust(right=0.85)
# 后续可直接调用fig.savefig保存或fig.show()展示
fig.show()

方案2：精准调整colorbar本身的内边距 #

如果需要更精细控制colorbar和主坐标轴的间距，可以先定位到colorbar对象，再修改其位置参数：

import module1 as m1
import matplotlib as mpl
from matplotlib.colorbar import Colorbar

with mpl.rc_context({'figure.figsize': (10,7), 'legend.borderaxespad': 4}):
    fig = m1.func_calc_and_draw(my_data)

# 遍历fig中的所有axes找到colorbar对象
for ax in fig.axes:
    if hasattr(ax, 'colorbar') and isinstance(ax.colorbar, Colorbar):
        cb = ax.colorbar
        # 获取当前colorbar的位置参数：[左边界位置, 下边界位置, 宽度, 高度]
        cb_pos = cb.ax.get_position()
        # 把colorbar向右移动0.05的相对距离，数值可根据需求自定义
        cb.ax.set_position([cb_pos.x0 + 0.05, cb_pos.y0, cb_pos.width, cb_pos.height])
        break

fig.show()

补充说明 #

如果原函数返回的fig里包含多个子图和多个colorbar，只需要在遍历的时候加上对应子图的判断条件，即可单独调整对应colorbar的位置。

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