如何让IPython Notebook代码等待用户点击图表后再执行？

问题场景 #

在Google Colab中配置ipympl后端后，已实现记录图表点击坐标的功能，但后续代码会直接执行，不会等待用户完成点击。尝试使用plt.waitforbuttonpress()时，程序会卡住，无法响应用户的图表点击操作。

配置代码：

!pip install ipympl
from google.colab import output
output.enable_custom_widget_manager()

原点击记录代码：

%matplotlib ipympl
import matplotlib.pyplot as plt

fig, ax = plt.subplots()

def onclick(event):
    ix, iy = event.xdata, event.ydata
    print(ix, iy)

cid = fig.canvas.mpl_connect('button_press_event', onclick)

解决方案 #

由于Colab的web交互式后端中，plt.waitforbuttonpress()会阻塞事件循环导致无响应，我们可以通过全局标记变量+循环休眠的方式实现等待，同时保证图表交互正常：

%matplotlib ipympl
import matplotlib.pyplot as plt
import time

# 绘制示例图表（必须有可点击内容，否则用户无交互对象）
fig, ax = plt.subplots()
ax.plot([1,2,3,4], [1,4,9,16])
ax.set_title("点击图表任意位置继续")

# 初始化标记变量
clicked = False
click_coords = []

def onclick(event):
    global clicked, click_coords
    # 确保点击在图表有效区域内
    if event.xdata is not None and event.ydata is not None:
        ix, iy = event.xdata, event.ydata
        click_coords.append((ix, iy))
        print(f"已记录点击坐标: {ix:.2f}, {iy:.2f}")
        clicked = True  # 标记点击完成

# 绑定点击事件
cid = fig.canvas.mpl_connect('button_press_event', onclick)

# 等待用户点击，循环休眠不阻塞事件循环
print("请点击图表中的点...")
while not clicked:
    time.sleep(0.1)  # 短休眠释放线程，让事件循环处理交互

# 解绑事件（可选，避免后续误触发）
fig.canvas.mpl_disconnect(cid)

# 后续代码，将在点击后执行
print('done!')
print(f"所有记录的坐标: {click_coords}")

原理说明 #

• plt.waitforbuttonpress()在Colab的web环境中会阻塞主线程，导致图表的交互事件无法被处理，因此出现无响应的情况。
• 循环+time.sleep(0.1)的方式，每次休眠时会释放线程资源，让后端的事件循环能够处理用户的点击操作，同时持续检查标记变量，直到用户完成点击后再执行后续代码。

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