plt figure中加入键盘鼠标互动

本文主要是介绍plt figure中加入键盘鼠标互动，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

如果想在plt.figure的图像上点击以获取图片中某点的坐标，
或者同时显示了几个subplot, 想通过键盘输入数字以选择第几个subplot,

这时就需要加入互动。

下面就以上case来说明如何互动。

1.鼠标点击获取坐标

定义一个onclick函数作为event触发。
这里假设最多可选3个坐标。
在close掉figure的时刻选点互动结束，点击的坐标就作为list的元素保存到coords里。

coords = []def onclick(event):global ix, iyix, iy = int(event.xdata), int(event.ydata)print(f'x = {ix}, y = {iy}')print("If finished, please close the figure.")global coordscoords.append((ix, iy))if len(coords) == 3:fig.canvas.mpl_disconnect(cid)print("finished! please close the figure.")return coordsfig = plt.figure()
plt.title("please click the targer, you can click at most 3 points.")
plt.axis("off")
ax = fig.add_subplot(111)
ax.imshow(img)
cid = fig.canvas.mpl_connect("button_press_event", onclick)
plt.show()input_point = np.array(coords) #list转np array

2.键盘输入

比如现在figure 中subplot 3张图片，想用键盘输入0,1,2，来选择到底用哪张图片。
这时定义onkey event。
close figure时选择结束。键盘输入会保存在select_id变量中。

def onkey(event):print('you selected id: ', event.key)global select_idselect_id = int(event.key)#fig.canvas.mpl_disconnect(cid)print("if finished, please close the figure.")return select_idfig = plt.figure()
plt.title("please select id, key input: 0/ 1/ 2")
plt.axis("off")
ax = fig.add_subplot(131)
ax.imshow(imgs[0])
ax = fig.add_subplot(132)
ax.imshow(imgs[1])
ax = fig.add_subplot(133)
ax.imshow(imgs[2])
cid = fig.canvas.mpl_connect('key_press_event', onkey)
plt.show()