Question

希望有人能在我的第一个程序中告诉我我做错了什么回调。

目标：

显示包含数据的图表
允许用户点击图4次。每次，X坐标被附加到已保存的列表中。
当鼠标移动时，其水平位置由垂直方向跟踪在情节内来回移动的线。（我保存了2D线对象为self.currentLine）
当用户通过点击选择一个点时，垂直线将被删除感兴趣的x坐标，并生成新的坐标以继续跟踪鼠标位置。

在用户输入结束时，应该有四条垂直线和类应该返回一个包含x坐标的列表。

目前，我无法找出更新线对象的正确方法情节（即我可以获得我想要的鼠标跟踪效果）。我也拿不到完成后返回值列表的类。

我知道while循环可能不是正确的方法，但我找不到合适的方法。

import matplotlib.pyplot as plt
import pdb

class getBval:
    def __init__(self):
        figWH = (8,5) # in
        self.fig = plt.figure(figsize=figWH)
        plt.plot(range(10),range(10),'k--')
        self.ax = self.fig.get_axes()[0]
        self.x = [] # will contain 4 "x" values
        self.lines = [] # will contain 2D line objects for each of 4 lines            

        self.connect =    self.ax.figure.canvas.mpl_connect
        self.disconnect = self.ax.figure.canvas.mpl_disconnect

        self.mouseMoveCid = self.connect("motion_notify_event",self.updateCurrentLine)
        self.clickCid     = self.connect("button_press_event",self.onClick)
    def updateCurrentLine(self,event):
        xx = [event.xdata]*2
        self.currentLine, = self.ax.plot(xx,self.ax.get_ylim(),'k')
        plt.show()
    def onClick(self, event):
        if event.inaxes:
            self.updateCurrentLine(event)
            self.x.append(event.xdata)
            self.lines.append(self.currentLine)
            del self.currentLine
            if len(self.x)==4:
                self.cleanup()
    def cleanup(self):
        self.disconnect(self.mouseMoveCid)
        self.disconnect(self.clickCid)
        return self


xvals = getBval()
print xvals.x

Answer 1

我建议您为垂直线使用Cursor窗口小部件，只收集点击次数的x值（而不是整个Line2D），这是一个示例：

import matplotlib.pyplot as plt
import matplotlib.widgets as mwidgets

class getBval:

    def __init__(self):
        figWH = (8,5) # in
        self.fig = plt.figure(figsize=figWH)
        plt.plot(range(10),range(10),'k--')
        self.ax = self.fig.get_axes()[0]
        self.x = [] # will contain 4 "x" values
        self.lines = [] # will contain 2D line objects for each of 4 lines            

        self.cursor = mwidgets.Cursor(self.ax, useblit=True, color='k')
        self.cursor.horizOn = False

        self.connect = self.ax.figure.canvas.mpl_connect
        self.disconnect = self.ax.figure.canvas.mpl_disconnect

        self.clickCid = self.connect("button_press_event",self.onClick)

    def onClick(self, event):
        if event.inaxes:
            self.x.append(event.xdata)
            if len(self.x)==4:
                self.cleanup()

    def cleanup(self):
        self.disconnect(self.clickCid)
        plt.close()


xvals = getBval()
plt.show()

print xvals.x

图形鼠标输入的回调 - 如何刷新图形，如何告诉Matplotlib我做完了？

1 个答案: