如何在Matplotlib中调整用于Blit的TransformedBbox背景区域的显示范围以实现高效缩放？

我完全懂你现在的困扰——用Matplotlib调整xlim和ylim来缩放时，每次都要重绘整个画布，速度慢得让人着急；而且用blit做局部更新时，之前保存的背景是基于整个轴的大小，没法跟着缩放范围调整，导致restore的时候要么显示不对，要么还是得全量重绘。其实咱们可以通过针对当前视图范围生成背景来解决这个问题，不用再依赖整个轴的bbox。

问题根源 #

你之前用的ax.bbox是整个轴的物理边界（画布像素坐标），不管你怎么调整xlim/ylim，这个bbox的大小和位置都不会变。所以当你缩放后，restore整个轴的背景，自然没法匹配新的视图范围，最后还是得调用cv.draw()重绘，这就是慢的核心原因。

解决方案：基于当前视图范围生成背景 #

我们需要把当前轴的数据视图范围（也就是xlim/ylim对应的区域）转换成画布上的像素bbox，然后只复制这个区域的背景。这样后续restore的时候，就能精准匹配缩放后的视图，不用全量重绘。

修改后的完整代码 #

import matplotlib.pyplot as plt
from matplotlib.axes import Axes
from matplotlib.backends.backend_agg import FigureCanvasAgg
from matplotlib.transforms import Bbox

fig, ax = plt.subplots(1, 1)
cv: FigureCanvasAgg = fig.canvas
ax: Axes

# 绘制初始曲线
ax.plot([1, 2, 3])
# 保存初始的轴范围，方便后续切换
orig_xlim = ax.get_xlim()
orig_ylim = ax.get_ylim()

bg = []
# 标记当前是否处于缩放状态
is_zoomed = [True]

def get_view_bbox(ax):
    """把当前轴的视图范围（数据坐标）转换成画布像素坐标的bbox"""
    view_lim = ax.viewLim
    # 构建数据坐标下的bbox
    data_bbox = Bbox.from_extents(view_lim.xmin, view_lim.ymin, view_lim.xmax, view_lim.ymax)
    # 转换为画布像素坐标的bbox
    return ax.transData.transform_bbox(data_bbox)

def toggle_zoom(event):
    global bg
    if is_zoomed[0]:
        # 切换到缩放状态
        ax.set_xlim(0.5, 1.5)
        ax.set_ylim(1, 2.5)
        is_zoomed[0] = False
    else:
        # 恢复初始视图
        ax.set_xlim(orig_xlim)
        ax.set_ylim(orig_ylim)
        is_zoomed[0] = True
    
    # 获取当前视图对应的画布bbox
    current_bbox = get_view_bbox(ax)
    
    # 第一次切换或恢复初始视图时，重新生成对应背景
    if not bg or is_zoomed[0]:
        # 用draw_idle替代draw，更高效
        cv.draw_idle()
        # 只复制当前视图区域的背景
        bg = [cv.copy_from_bbox(current_bbox)]
    else:
        # 直接恢复之前保存的缩放背景
        cv.restore_region(bg[0])
        # 只更新当前视图对应的画布区域
        cv.blit(current_bbox)
    
    # 刷新画布事件
    cv.flush_events()

# 绑定鼠标点击事件切换缩放
fig.canvas.mpl_connect('button_press_event', toggle_zoom)
plt.show()

关键改动说明 #

1. 新增get_view_bbox函数：这个函数的作用是把当前轴的视图范围（数据坐标）转换成画布上的像素区域，这样我们复制的背景就只包含当前可见的部分，而不是整个轴。
2. 替换ax.bbox为视图对应的bbox：不再保存整个轴的背景，只保存当前视图范围内的背景，这样restore的时候就能精准匹配缩放后的显示区域。
3. 用draw_idle替代draw：draw_idle()会等待系统空闲时再绘制，避免不必要的重绘操作，进一步提升效率。
4. 只blit需要更新的区域：切换到缩放状态时，直接restore之前保存的缩放背景，然后只更新当前视图对应的画布区域，不用全量重绘整个画布。

这样调整后，你会发现缩放的速度明显提升，因为大部分时候都不需要重绘整个画布，只做局部的blit更新就够了。

备注：内容来源于stack exchange，提问作者white.seolpyo.com