基于Tkinter Matplotlib画布的两点间动态散点图实现需求

核心思路 #

• 新增状态变量跟踪是否已选中第一个散点，以及存储第一个点的坐标
• 绑定鼠标移动事件，在选中第一个点后，实时用鼠标当前坐标作为临时第二个点，重新计算并更新中间散点
• 点击第二个目标散点时，固定临时点为目标点，结束动态跟踪

具体代码修改 #

假设你的现有代码已经定义了scatter_pts（橙色散点）、middle_pts（蓝色中间点）、canvas（Matplotlib画布对象）、ax（绘图轴对象），按以下步骤修改：

1. 添加状态变量：

# 初始化状态：是否已选中第一个点，以及第一个点的坐标
selected_first = False
first_point = None
# 用于临时显示的虚拟第二个点
temp_scatter = None

2. 修改拾取事件处理函数：

def on_pick(event):
    global selected_first, first_point, temp_scatter
    # 获取点击的散点索引与坐标
    ind = event.ind[0]
    pt = scatter_pts.get_offsets()[ind]
    
    if not selected_first:
        # 选中第一个点，初始化临时虚拟点
        selected_first = True
        first_point = pt
        temp_scatter = ax.scatter(pt[0], pt[1], color='red', marker='o')
    else:
        # 选中第二个点，固定中间点并清除临时点
        selected_first = False
        update_middle_points(first_point, pt)
        temp_scatter.remove()
        temp_scatter = None
        first_point = None
    canvas.draw_idle()

3. 添加鼠标移动事件处理函数：

def on_motion(event):
    global selected_first, first_point, temp_scatter, middle_pts
    if not selected_first or event.inaxes != ax:
        return
    # 获取鼠标当前坐标（过滤无效坐标）
    curr_x, curr_y = event.xdata, event.ydata
    if curr_x is None or curr_y is None:
        return
    
    # 更新临时虚拟点位置
    temp_scatter.set_offsets([[curr_x, curr_y]])
    # 实时计算并更新中间散点
    update_middle_points(first_point, (curr_x, curr_y))
    canvas.draw_idle()

# 绑定鼠标移动事件到画布
canvas.mpl_connect('motion_notify_event', on_motion)

4. 实现通用的中间点更新函数：

def update_middle_points(pt1, pt2):
    x1, y1 = pt1
    x2, y2 = pt2
    # 按线性插值生成4个中间点
    middle_coords = [
        (x1 + (x2-x1)*i/5, y1 + (y2-y1)*i/5)
        for i in range(1,5)
    ]
    middle_pts.set_offsets(middle_coords)

关键说明 #

• 使用draw_idle()代替draw()，减少不必要的画布刷新，提升动态效果流畅度
• 仅在选中第一个点且鼠标处于绘图区域内时，才触发动态更新逻辑
• 临时虚拟点用红色区分，方便用户识别当前跟踪状态

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