使用Matplotlib Colorbar时如何保持图形尺寸恒定

问题核心：当使用tight_layout和bbox_inches='tight'保存图表时，Colorbar的刻度标签因数据小数位数/量级不同，占用宽度变化，导致主绘图区和整体图表尺寸不一致。以下是两种可靠的解决方法：

方法一：固定Colorbar的宽度（推荐） #

使用mpl_toolkits.axes_grid1.make_axes_locatable工具，为主绘图轴分割出固定宽度的区域放置Colorbar，彻底避免标签长度影响布局。

示例代码：

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable
import numpy as np

def plot_fixed_cbar(order):
    fig, ax = plt.subplots(figsize=(6, 4))
    # 生成对应量级的随机数据
    data = np.random.rand(10, 10) * 10**order
    
    im = ax.imshow(data)
    
    # 分割主轴，创建固定宽度的Colorbar轴
    divider = make_axes_locatable(ax)
    # 可设置百分比（如"5%"）或固定像素（如"20px"）
    cax = divider.append_axes("right", size="5%", pad=0.1)
    
    # 添加Colorbar到固定轴
    cbar = fig.colorbar(im, cax=cax)
    # 可选：统一科学计数法显示规则
    cbar.formatter.set_powerlimits((-3, 3))
    cbar.update_ticks()
    
    plt.tight_layout()
    plt.savefig(f"fixed_cbar_order_{order}.png", bbox_inches='tight')
    plt.close()

# 生成不同量级的图表
plot_fixed_cbar(-3)
plot_fixed_cbar(0)

方法二：统一Colorbar刻度标签格式 #

已知数据最多3位小数，强制所有Colorbar的刻度标签显示3位小数，让标签宽度保持一致，从而稳定布局。

示例代码：

import matplotlib.pyplot as plt
from matplotlib.ticker import FormatStrFormatter
import numpy as np

def plot_uniform_cbar_format(order):
    fig, ax = plt.subplots(figsize=(6, 4))
    data = np.random.rand(10, 10) * 10**order
    
    im = ax.imshow(data)
    cbar = fig.colorbar(im)
    
    # 强制刻度显示3位小数
    cbar.formatter = FormatStrFormatter('%.3f')
    # 若需科学计数法，可替换为以下代码：
    # from matplotlib.ticker import ScalarFormatter
    # formatter = ScalarFormatter(useMathText=True)
    # formatter.set_scientific(True)
    # formatter.set_precision(3)
    # cbar.formatter = formatter
    cbar.update_ticks()
    
    plt.tight_layout()
    plt.savefig(f"uniform_format_order_{order}.png", bbox_inches='tight')
    plt.close()

plot_uniform_cbar_format(-3)
plot_uniform_cbar_format(0)

注意事项 #

• 方法一不受标签长度影响，即使标签过长也只会在固定宽度内换行或调整字号，主绘图区尺寸完全稳定。
• 方法二更简洁，但如果数据量级差异极大（如1e-3到1e3），科学计数法的指数部分可能仍会占用不同空间，此时方法一更稳妥。

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