将 %matplotlib notebook 的函数转换为 Python 脚本

Question

我正在尝试将几个 python 笔记本文件转换为 python 脚本，以便我可以创建一个 python 包... 我现在遇到的问题是在 python 笔记本中，有

%matplotlib notebook

使用户可以单击绘图并接收他/她刚刚单击的绘图的坐标的魔术按钮。实现方式如下：

f = urllib.urlopen(url_link)
fig = plt.figure(figsize=(3,3))
coords = coordinates(my_coord)[1]
image = plt.imread(f)
plt.imshow(image)

collector = {'x':0,'y':0}
def onclick(event):
    collector['x'] = event.x
    collector['y'] = event.y

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

现在我想在 python 脚本中创建相同的函数。但在脚本中

%matplotlib notebook

不工作。有没有其他方法可以让我的 python 脚本做同样的事情：显示绘图和用户可以点击并返回用户点击的坐标？

提前谢谢！

Answer 1

原则上不需要这样的事件处理，因为这在 matplotlib 绘图窗口中自动可用。

以下脚本

import matplotlib.pyplot as plt

image = plt.imread("image.jpg")

fig, ax = plt.subplots(figsize=(4,3))
ax.imshow(image)

plt.show()

产生这个窗口

在下方状态栏中显示当前鼠标位置的坐标。

但当然可以通过使用相同的事件处理机制来扩展功能，就像问题中一样

import matplotlib.pyplot as plt

image = plt.imread("image.jpg")

fig, ax = plt.subplots(figsize=(4,3))
ax.imshow(image)

collector = {'x':0,'y':0}
def onclick(event):
    collector['x'] = event.x
    collector['y'] = event.y
    print(collector)

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

plt.show()

请注意，这将打印图形坐标，而不是数据坐标。由于图形坐标在大多数情况下是无用的，因此最好使用数据坐标，像这样。

collector['x'] = event.xdata
collector['y'] = event.ydata

另请参阅 matplotlib 页面上的 event handling article。