如何在matplotlib中实现右键点击拖拽平移绘图功能

实现思路 #

matplotlib内置的导航工具栏默认绑定左键为平移触发键，我们可以通过修改工具栏按键配置、手动绑定鼠标事件两种方式，把平移触发逻辑改成右键，完美适配PySimpleGUI调用Tkinter的底层环境。

完整实现代码 #

import PySimpleGUI as sg
import matplotlib.pyplot as plt
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2Tk
import numpy as np

# 配置右键平移逻辑
def set_right_click_pan(fig, canvas):
    # 平移状态记录
    pan_state = {
        "is_panning": False,
        "last_x": 0,
        "last_y": 0,
        "xlim": None,
        "ylim": None
    }
    
    def on_press(event):
        # 仅响应右键且点击位置在绘图区域内的操作
        if event.button == 3 and event.inaxes:
            pan_state["is_panning"] = True
            pan_state["last_x"] = event.x
            pan_state["last_y"] = event.y
            pan_state["xlim"] = event.inaxes.get_xlim()
            pan_state["ylim"] = event.inaxes.get_ylim()
            canvas.config(cursor="fleur")
    
    def on_motion(event):
        if not pan_state["is_panning"] or not event.inaxes:
            return
        dx = event.x - pan_state["last_x"]
        dy = event.y - pan_state["last_y"]
        if dx == 0 and dy == 0:
            return
        # 像素位移转坐标位移
        ax = event.inaxes
        x_pixel_range = ax.get_window_extent().width
        y_pixel_range = ax.get_window_extent().height
        x_range = pan_state["xlim"][1] - pan_state["xlim"][0]
        y_range = pan_state["ylim"][1] - pan_state["ylim"][0]
        dx_coord = dx * x_range / x_pixel_range
        dy_coord = dy * y_range / y_pixel_range
        # 更新坐标轴范围
        ax.set_xlim(pan_state["xlim"][0] - dx_coord, pan_state["xlim"][1] - dx_coord)
        ax.set_ylim(pan_state["ylim"][0] + dy_coord, pan_state["ylim"][1] + dy_coord)
        canvas.draw_idle()
    
    def on_release(event):
        if event.button == 3:
            pan_state["is_panning"] = False
            canvas.config(cursor="arrow")
    
    # 绑定鼠标事件
    canvas.mpl_connect("button_press_event", on_press)
    canvas.mpl_connect("motion_notify_event", on_motion)
    canvas.mpl_connect("button_release_event", on_release)

# PySimpleGUI画布嵌入逻辑
def draw_figure(canvas, figure):
    figure_canvas_agg = FigureCanvasTkAgg(figure, canvas)
    figure_canvas_agg.draw()
    figure_canvas_agg.get_tk_widget().pack(side="top", fill="both", expand=1)
    # 不需要原生工具栏可以注释下面两行
    toolbar = NavigationToolbar2Tk(figure_canvas_agg, canvas)
    toolbar.update()
    return figure_canvas_agg

# 测试样例
if __name__ == "__main__":
    # 生成测试绘图
    fig, ax = plt.subplots()
    x = np.linspace(0, 10, 1000)
    ax.plot(x, np.sin(x) * np.cos(2*x))
    ax.set_title("右键拖拽平移测试")
    # 构建界面
    layout = [
        [sg.Canvas(key="-CANVAS-")],
        [sg.Button("退出")]
    ]
    window = sg.Window("Matplotlib右键平移Demo", layout, finalize=True)
    # 嵌入画布并启用右键平移
    canvas = draw_figure(window["-CANVAS-"].TKCanvas, fig)
    set_right_click_pan(fig, canvas)
    # 事件循环
    while True:
        event, values = window.read()
        if event in (sg.WIN_CLOSED, "退出"):
            break
    window.close()

关键说明 #

• 上述实现不依赖matplotlib原生工具栏，可单独使用，也可以和原生工具栏共存。如果要保留原生工具栏的平移按钮功能，只需要额外添加一行toolbar._pan_btn = 3修改工具栏的默认触发按键即可。
• 代码中event.button == 3对应Tkinter体系下的鼠标右键，无需额外适配即可在你的PySimpleGUI+Tkinter环境中运行。
• 平移逻辑兼容线性轴、对数轴等所有matplotlib支持的坐标轴类型。

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