Matplotlib创建依赖Y轴缩放的自适应圆形混合变换问题

问题描述 #

需要创建一个满足以下要求的圆形：

• 显示形状不受轴缩放影响（始终保持圆形）
• 位置基于数据坐标
• 半径依赖Y轴的缩放比例

参考Matplotlib的transforms教程及物理坐标绘图内容构建了变换流水线，但无法合理定义fig_to_data_scaler。明确需要混合变换（blended transformation），结合ax.transData的Y轴缩放与fig.dpi_scale_trans（可能需反转），同时X轴不受数据变换影响。

尝试多种变换组合后，其中一个方案能正确放置圆形且平移正常，但缩放轴时圆形大小不更新，需定位问题原因。

尝试的代码片段 #

初始方案 #

from matplotlib import pyplot as plt, patches as mpatch, transforms as mtrans

fig, ax = plt.subplots()
x, y = 5, 10
r = 3
transform = fig.dpi_scale_trans + fig_to_data_scaler + mtrans.ScaledTranslation(x, y, ax.transData)
ax.add_patch(mpatch.Circle((0, 0), r, edgecolor='k', linewidth=2, facecolor='w', transform=t))

后续尝试方案 #

vertical_scale_transform = mtrans.AffineDeltaTransform(ax.transData)
reflection = mtrans.Affine2D.from_values(0, 1, 1, 0, 0, 0)
uniform_scale_transform = mtrans.blended_transform_factory(reflection + vertical_scale_transform + reflection, vertical_scale_transform)
t = uniform_scale_transform + mtrans.ScaledTranslation(x, y, ax.transData)
ax.add_patch(mpatch.Circle((0, 0), r, edgecolor='k', linewidth=2, facecolor='w', transform=t))

问题原因分析 #

核心问题在于变换未绑定轴的动态更新机制，或变换流水线未正确包含轴缩放的实时反馈：

1. 初始方案中fig_to_data_scaler未定义，缺失了物理坐标到数据坐标的关键转换环节，导致轴缩放时无法同步更新圆形大小。
2. 后续方案里的AffineDeltaTransform是基于创建时ax.transData的快照构建的，仅记录初始变换增量，不会随轴缩放实时刷新，因此圆形大小始终保持初始状态。

正确实现方案 #

要满足需求，需构建动态混合变换，让X轴保持数据坐标定位，Y轴兼顾数据坐标定位和缩放比例控制半径，同时确保变换随轴缩放实时更新：

from matplotlib import pyplot as plt, patches as mpatch, transforms as mtrans

fig, ax = plt.subplots()
x_data, y_data = 5, 10
radius_y = 3  # 半径对应的Y轴数据单位

# 实时获取Y轴数据到物理坐标的缩放变换
def get_y_scale_transform():
    trans_data = ax.transData.get_matrix()
    # 提取Y轴缩放因子，转换为物理单位缩放
    return mtrans.Affine2D.from_scale(1, trans_data[1,1]) + ax.transData.inverted() + fig.dpi_scale_trans.inverted()

# 绑定轴事件，实时更新变换
def update_transform(event):
    blended_trans = mtrans.blended_transform_factory(
        ax.transData,  # X轴：数据坐标定位
        mtrans.ScaledTranslation(x_data, y_data, ax.transData) + get_y_scale_transform()
    )
    circle.set_transform(blended_trans)
    fig.canvas.draw_idle()

# 创建圆形并添加到轴
circle = mpatch.Circle((0, 0), radius_y, edgecolor='k', linewidth=2, facecolor='w')
ax.add_patch(circle)

# 绑定轴缩放、 resize事件
fig.canvas.mpl_connect('resize_event', update_transform)
fig.canvas.mpl_connect('xlim_changed', update_transform)
fig.canvas.mpl_connect('ylim_changed', update_transform)

# 初始化轴范围
ax.set_xlim(0, 10)
ax.set_ylim(0, 20)
plt.show()

实现说明 #

• get_y_scale_transform()实时提取当前Y轴的缩放因子，将数据单位转换为物理单位，保证圆形半径随Y轴缩放同步变化。
• 通过绑定resize_event、xlim_changed、ylim_changed事件，每次轴状态变化时重新计算变换，确保圆形大小实时响应。
• 混合变换让X轴始终基于数据坐标定位，Y轴同时实现位置定位和缩放驱动的半径控制，圆形形状保持不变。

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