如何基于ylabel的x坐标与title的y坐标为Matplotlib子图设置角标？

问题背景 #

需要在Matplotlib多面板图中实现以下需求：

• 先用fig.align_ylabels()统一对齐所有子图的y轴标签
• 基于y轴标签的x坐标、子图标题的y坐标自动放置子图角标（如“1”“2”），确保所有角标对齐

尝试现有方法时遇到的问题：

• ax.yaxis.get_label_position()仅返回left/right，无法获取具体坐标值
• ax.yaxis.get_label().get_position()和ax.title.get_position()返回的是轴内分数坐标，测试中两个子图返回值均为(0,1.0)，无法区分实际渲染位置，导致角标无法对齐

测试代码：

import matplotlib.pyplot as plt
fig,axs=plt.subplots(1,2,figsize=[6,3])
for i,ax in enumerate(axs):
    ax.set(title='title',xlabel='xlabel',ylabel='ylabel',xlim=[0,10],ylim=[0,10])
    x,y=ax.yaxis.get_label().get_position()[0],ax.title.get_position()[1]
    ax.text(x=x,y=y,s=i+1,ha='left',va='center',transform=ax.transAxes)
    print(f"for subplot#{i+1} coordinates of the label, x,y={x},{y}",)

输出：

for subplot#1 coordinates of the label, x,y=0,1.0
for subplot#2 coordinates of the label, x,y=0,1.0

解决方案 #

核心是将元素的屏幕坐标转换为figure归一化坐标系，基于实际渲染位置确定角标坐标：

修改后的代码 #

import matplotlib.pyplot as plt

fig, axs = plt.subplots(1, 2, figsize=[6, 3])
# 先设置所有子图基础内容
for i, ax in enumerate(axs):
    ax.set(title='标题', xlabel='x轴', ylabel='y轴', xlim=[0,10], ylim=[0,10])

# 统一对齐y轴标签
fig.align_ylabels(axs)

# 遍历子图添加对齐的角标
for i, ax in enumerate(axs):
    # 获取y轴标签的屏幕边界框，转换为figure归一化坐标
    ylabel_bbox = ax.yaxis.get_label().get_window_extent().transformed(fig.transFigure.inverted())
    # 获取标题的屏幕边界框，转换为figure归一化坐标
    title_bbox = ax.title.get_window_extent().transformed(fig.transFigure.inverted())
    
    # 角标x坐标取y轴标签左边界，y坐标取标题上边界
    corner_x = ylabel_bbox.x0
    corner_y = title_bbox.y1
    
    # 在figure坐标系下添加角标
    fig.text(corner_x, corner_y, str(i+1), ha='right', va='center', fontweight='bold')

plt.tight_layout()
plt.show()

关键步骤说明 #

1. 先执行y轴标签对齐：必须在获取标签坐标前调用fig.align_ylabels()，否则标签位置未完成统一调整
2. 坐标转换逻辑：
   • get_window_extent()获取元素的屏幕像素边界框
   • transformed(fig.transFigure.inverted())将屏幕坐标转换为figure归一化坐标（范围0-1，对应figure的宽高比例）
3. 角标定位规则：取y轴标签的左边界x值、标题的上边界y值，确保角标在y轴标签左侧、标题高度位置，实现所有子图角标统一对齐

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