如何通过鼠标事件实现Matplotlib中图像的可拉伸功能？

嘿，我懂你想要的效果——让那个已经能拖拽的图像还能通过鼠标拖动来放大缩小对吧？我给你调整了代码，加上了鼠标拉伸的逻辑，你直接用就行：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.image as image
from matplotlib.offsetbox import OffsetImage, AnnotationBbox

fig = plt.figure()
X = [1, 2, 3, 4, 5, 6, 7]
Y = [1, 3, 4, 2, 5, 8, 6]
mainaxes = fig.add_axes([0.1, 0.1, 0.8, 0.8])  # 主坐标轴

# 读取图像
img = image.imread('https://upload.wikimedia.org/wikipedia/commons/7/70/Example.png')
initial_zoom = 0.6
imagebox = OffsetImage(img, zoom=initial_zoom)
imgbox = AnnotationBbox(imagebox, (0.3, 0.5), frameon=True)
mainaxes.add_artist(imgbox)
imgbox.draggable()  # 保留原有的拖拽功能

# 用于记录鼠标拉伸的状态
press_info = None

def on_press(event):
    """记录鼠标按下时的位置和当前的缩放比例"""
    global press_info
    # 只在点击到图像框的时候触发
    if event.inaxes == mainaxes and imgbox.contains(event)[0]:
        press_info = {
            'x': event.x,
            'y': event.y,
            'current_zoom': imagebox.get_zoom()
        }

def on_drag(event):
    """拖动鼠标时调整图像的缩放比例"""
    global press_info
    if press_info is None:
        return
    # 计算鼠标移动的垂直距离（用y轴变化更符合直觉，上下拖动缩放）
    delta_y = event.y - press_info['y']
    # 调整缩放比例，这里的系数可以自己调，控制缩放灵敏度
    zoom_change = delta_y * 0.001
    new_zoom = press_info['current_zoom'] + zoom_change
    # 限制缩放范围，避免过度缩放
    new_zoom = max(0.1, min(new_zoom, 2.0))
    imagebox.set_zoom(new_zoom)
    # 更新图像框并重绘
    imgbox.update_artist(fig)
    fig.canvas.draw_idle()

def on_release(event):
    """释放鼠标时重置状态"""
    global press_info
    press_info = None

# 绑定鼠标事件到figure
fig.canvas.mpl_connect('button_press_event', on_press)
fig.canvas.mpl_connect('motion_notify_event', on_drag)
fig.canvas.mpl_connect('button_release_event', on_release)

plt.show()

关键逻辑说明： #

• 鼠标按下事件：当点击到图像框时，记录当前的鼠标位置和图像的缩放比例，这样拖动时能基于初始值计算变化
• 鼠标拖动事件：根据鼠标垂直移动的距离来调整缩放比例，这里用y轴移动是因为上下拖动比左右更符合我们对缩放的直觉，你也可以改成x轴或者两者结合
• 缩放范围限制：加了max(0.1, min(new_zoom, 2.0))来防止图像缩得太小或者大到离谱，你可以根据需求调整这个范围
• 重绘更新：每次调整缩放后调用draw_idle()来刷新图像，保证界面实时更新

现在你运行这个代码，就能按住图像拖动来移动它，还能按住鼠标上下拖动来放大缩小啦！

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