如何重写Matplotlib矩形缩放实现MATLAB式单维度缩放？

完全可以实现这个需求，核心思路是重写矩形选框的缩放逻辑，根据选框的宽高比自动判断单轴还是双轴缩放，以下是具体实现方案：

核心代码实现 #

先看完整的可运行示例：

import matplotlib.pyplot as plt
from matplotlib.widgets import RectangleSelector
import numpy as np

def on_select(eclick, erelease):
    # 获取选框的坐标范围
    x1, y1 = eclick.xdata, eclick.ydata
    x2, y2 = erelease.xdata, erelease.ydata
    x_left, x_right = sorted([x1, x2])
    y_bottom, y_top = sorted([y1, y2])
    
    # 计算选框宽高比，阈值可根据习惯调整
    width = x_right - x_left
    height = y_top - y_bottom
    
    ax = eclick.inaxes
    if width / height > 3:
        # 细长水平选框 → 仅X轴缩放
        ax.set_xlim(x_left, x_right)
    elif height / width > 3:
        # 细长垂直选框 → 仅Y轴缩放
        ax.set_ylim(y_bottom, y_top)
    else:
        # 普通比例选框 → 双轴正常缩放
        ax.set_xlim(x_left, x_right)
        ax.set_ylim(y_bottom, y_top)
    plt.draw()

# 生成测试数据
x = np.linspace(0, 10, 100)
y = np.sin(x) + np.random.randn(100)*0.1

fig, ax = plt.subplots()
ax.plot(x, y)

# 创建自定义矩形选择器，绑定左键绘制选框
rs = RectangleSelector(ax, on_select, 
                       drawtype='box', useblit=True,
                       button=[1],  # 仅响应左键操作
                       minspanx=5, minspany=5,  # 最小选框像素，避免误触
                       spancoords='pixels',
                       interactive=True)

plt.show()

代码说明 #

• 选框判断逻辑：通过计算选框的宽高比（阈值设为3，可自行调整），区分水平细长、垂直细长和普通选框，分别执行对应缩放操作。
• RectangleSelector：替代默认的缩放工具，捕获鼠标绘制矩形的事件，触发自定义缩放逻辑。
• 操作方式：直接用左键绘制选框即可——画水平窄条就只缩X轴，画垂直窄条就只缩Y轴，普通矩形则双轴缩放，完全贴合MATLAB的使用习惯。

进阶：替换工具栏默认缩放按钮 #

如果希望点击工具栏的缩放按钮来激活这个功能，可自定义工具栏类：

from matplotlib.backend_bases import NavigationToolbar2

class CustomToolbar(NavigationToolbar2):
    def zoom(self, *args):
        # 切换自定义缩放工具的激活状态
        if not hasattr(self, 'rect_selector'):
            self.rect_selector = RectangleSelector(self.canvas.figure.axes[0], on_select,
                                                   drawtype='box', useblit=True,
                                                   button=[1],
                                                   minspanx=5, minspany=5,
                                                   spancoords='pixels',
                                                   interactive=True)
        else:
            self.rect_selector.set_active(not self.rect_selector.get_active())

# 替换默认工具栏
fig.canvas.toolbar = CustomToolbar(fig.canvas)

这样点击工具栏的缩放按钮后，就能用上述自定义逻辑进行缩放，和原生操作流程无缝衔接。

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