如何修改Matplotlib的Zoom to Rectangle按钮使其始终缩放到正方形？

你可以通过自定义Matplotlib的矩形缩放逻辑，让用户选择的区域自动调整为符合当前轴纵横比的正方形，替代默认的"Zoom to Rectangle"工具。

核心思路 #

捕获用户的矩形选择事件，计算选择区域的中心，然后根据当前轴的纵横比，将选择范围调整为正方形，再设置新的坐标轴范围。

代码实现 #

基础版本（直接替换缩放行为） #

import matplotlib.pyplot as plt
from matplotlib.widgets import RectangleSelector
from matplotlib.backend_bases import MouseButton

def square_zoom(event_click, event_release):
    ax = plt.gca()
    # 获取用户点击和释放的坐标
    x1, y1 = event_click.xdata, event_click.ydata
    x2, y2 = event_release.xdata, event_release.ydata

    if None in (x1, y1, x2, y2):
        return

    # 计算选择区域的中心
    cx, cy = (x1 + x2)/2, (y1 + y2)/2
    # 计算x、y方向的选择长度
    dx, dy = abs(x2 - x1), abs(y2 - y1)

    # 获取当前轴的纵横比
    aspect = ax.get_aspect()
    if aspect == 'equal':
        # 等比例下取较小的边长作为正方形边长
        side = min(dx, dy)
    else:
        # 非等比例时转换为实际长度比例
        aspect_ratio = float(aspect)
        side = min(dx, dy * aspect_ratio)

    # 计算正方形的边界
    new_x1, new_x2 = cx - side/2, cx + side/2
    if aspect == 'equal':
        new_y1, new_y2 = cy - side/2, cy + side/2
    else:
        new_y1, new_y2 = cy - (side/2)/aspect_ratio, cy + (side/2)/aspect_ratio

    # 更新坐标轴范围
    ax.set_xlim(new_x1, new_x2)
    ax.set_ylim(new_y1, new_y2)
    plt.draw()

# 创建示例图形
fig, ax = plt.subplots()
ax.plot([1,2,3,4], [1,4,9,16])
ax.set_aspect('equal')

# 绑定自定义矩形选择器，左键触发
rs = RectangleSelector(
    ax, square_zoom,
    drawtype='box', useblit=True,
    button=[MouseButton.LEFT],
    minspanx=5, minspany=5,
    spancoords='pixels'
)

plt.show()

进阶版本（替换导航栏的默认Zoom按钮） #

如果你想保留Matplotlib导航栏的交互，替换默认的"Zoom to Rectangle"按钮行为：

import matplotlib.pyplot as plt
from matplotlib.widgets import RectangleSelector
from matplotlib.backend_bases import MouseButton
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT, FigureCanvasQTAgg
from PyQt5.QtWidgets import QMainWindow, QApplication

def square_zoom(event_click, event_release):
    ax = plt.gca()
    x1, y1 = event_click.xdata, event_click.ydata
    x2, y2 = event_release.xdata, event_release.ydata

    if None in (x1, y1, x2, y2):
        return

    cx, cy = (x1 + x2)/2, (y1 + y2)/2
    dx, dy = abs(x2 - x1), abs(y2 - y1)

    aspect = ax.get_aspect()
    if aspect == 'equal':
        side = min(dx, dy)
    else:
        aspect_ratio = float(aspect)
        side = min(dx, dy * aspect_ratio)

    new_x1, new_x2 = cx - side/2, cx + side/2
    if aspect == 'equal':
        new_y1, new_y2 = cy - side/2, cy + side/2
    else:
        new_y1, new_y2 = cy - (side/2)/aspect_ratio, cy + (side/2)/aspect_ratio

    ax.set_xlim(new_x1, new_x2)
    ax.set_ylim(new_y1, new_y2)
    plt.draw()

class CustomNavToolbar(NavigationToolbar2QT):
    def __init__(self, canvas, parent=None):
        super().__init__(canvas, parent)
        # 找到默认Zoom按钮，替换其触发事件
        for action in self.actions():
            if action.text() == 'Zoom to Rectangle':
                action.triggered.disconnect()
                action.triggered.connect(self.enable_square_zoom)
                break

    def enable_square_zoom(self):
        # 激活自定义矩形选择器
        rs.set_active(True)

# 创建图形和选择器
fig, ax = plt.subplots()
ax.plot([1,2,3,4], [1,4,9,16])
ax.set_aspect('equal')

rs = RectangleSelector(
    ax, square_zoom,
    drawtype='box', useblit=True,
    button=[MouseButton.LEFT],
    minspanx=5, minspany=5,
    spancoords='pixels'
)
rs.set_active(False)  # 默认不激活

# 启动Qt窗口
app = QApplication([])
window = QMainWindow()
canvas = FigureCanvasQTAgg(fig)
toolbar = CustomNavToolbar(canvas, window)
window.addToolBar(toolbar)
window.setCentralWidget(canvas)
window.show()
app.exec_()

说明 #

• 基础版本直接用自定义矩形选择器替代默认缩放，适合简单场景。
• 进阶版本修改了导航工具栏的默认按钮，保留了原生界面的交互逻辑。
• 代码会根据当前轴的aspect参数自动调整正方形比例，无论是equal还是自定义比例都能适配。

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