如何通过图形点击交互实现时间历程拆分并返回拆分点x轴值

不需要引入Plotly、Dash这类重型Web框架，matplotlib原生支持阻塞式交互，可直接将用户点击的坐标返回给上层调用代码，无需额外处理跨组件回调传值。最简实现可直接使用plt.ginput()接口完成，代码如下：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

def split_TH(df: pd.DataFrame, y_col: str = "data") -> tuple[pd.DataFrame, pd.DataFrame]:
    fig, ax = plt.subplots()
    ax.plot(df.index, df[y_col])
    ax.set_title("点击拆分位置后按回车/右键确认")
    
    # 阻塞等待用户完成1次点击，返回点击坐标
    click_res = plt.ginput(n=1, timeout=-1, show_clicks=True)
    plt.close(fig)
    
    if not click_res:
        raise ValueError("未选择有效拆分点")
    
    split_x = click_res[0][0]
    # 匹配最近的整数索引，做边界保护
    split_idx = int(round(split_x))
    split_idx = max(0, min(split_idx, len(df)))
    
    return df.iloc[:split_idx].copy(), df.iloc[split_idx:].copy()

# 测试
if __name__ == "__main__":
    df = pd.DataFrame({"data": np.random.rand(100)})
    part1, part2 = split_TH(df)
    print(f"拆分位置索引：{len(part1)}，前半段长度{len(part1)}，后半段长度{len(part2)}")

使用说明 #

• 调用函数后会弹出matplotlib绘图窗口，在需要拆分的位置点击鼠标左键，点击位置会显示红色标记方便校准
• 选好位置后按回车键或鼠标右键即可确认，窗口自动关闭后直接返回拆分完成的两个DataFrame
• plt.ginput()会暂停当前代码执行直到用户完成操作，返回值直接传递回上层调用逻辑，不存在回调结果无法传出的问题
• 如果需要精准匹配距离点击位置最近的真实数据点，可将索引计算逻辑替换为split_idx = np.argmin(np.abs(df.index.to_numpy() - split_x))
• 代码内置了边界校验，即使用户点到绘图区域外，也不会触发索引越界错误

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