Matplotlib 事件选择器 - 类内部

Question

我正在尝试在类中插入此 python 文档中显示的事件选择器示例

http://matplotlib.org/users/event_handling.html

代码是这样的

import numpy as np
import matplotlib.pyplot as plt


class Test:
    def __init__(self,line):
        self.line = line
        self.cidpress = self.line.figure.canvas.mpl_connect('button_press_event', self.onpick)

    def onpick(self, event):
        thisline = event.artist
        xdata = thisline.get_xdata()
        ydata = thisline.get_ydata()
        ind = event.ind
        points = tuple(zip(xdata[ind], ydata[ind]))
        print('onpick points:', points)


fig = plt.figure()
ax = fig.add_subplot(111)
ax.set_title('click on points')

line, = ax.plot(np.random.rand(10), 'o', picker=5)  # 5 points tolerance
a = Test(line)

plt.show()

但是当鼠标点击一个点时我得到这个错误。

AttributeError: 'MouseEvent' object has no attribute 'artist'

这可能是什么原因？ 当不在类内时，代码可以完美运行

非常感谢

Answer 1

我怀疑代码在类之外是否有效。您在这里面临的问题是您使用了'button_press_event'，它没有artist 属性。无论在课堂内还是课堂外，这都不会改变。

• 如果您想使用event.artist，您需要使用'pick_event'。这显示在 matplotlib 页面上的 event picking example 中。

• 如果要使用'button_press_event'，则不能使用event.artist，而是需要通过查询某个艺术家是否包含该事件来找出被点击的元素，例如if line.contains(event)[0]: ...。参见例如这个问题：How to check if click is on scatter plot point with multiple markers (matplotlib)