Question

关于this matplotlib示例，该示例绘制了可拖动矩形，我尝试对多边形进行相同操作。

到目前为止，我已经能够绘制多边形并将其拖动到画布上的某个位置。如果释放鼠标按钮，将无法再次移动多边形，这就是我的问题。我想尽可能多地通过每次鼠标拖动来拖放多边形。

我还注意到，在移动多边形之后，我仍然可以单击多边形以前的位置，然后再次拖动它。因此，初始import matplotlib.pyplot as plt from matplotlib.patches import Polygon #from matplotlib.collections import PatchCollection class DraggablePolygon: lock = None def __init__(self, polygon): self.poly = polygon self.press = None def connect(self): 'connect to all the events we need' self.cidpress = self.poly.figure.canvas.mpl_connect( 'button_press_event', self.on_press) self.cidrelease = self.poly.figure.canvas.mpl_connect( 'button_release_event', self.on_release) self.cidmotion = self.poly.figure.canvas.mpl_connect( 'motion_notify_event', self.on_motion) def on_press(self, event): 'on button press we will see if the mouse is over us and store some data' if event.inaxes != self.poly.axes: return if DraggablePolygon.lock is not None: return contains, attrd = self.poly.contains(event) if not contains: return x0, y0 = geometry[0] self.press = x0, y0, event.xdata, event.ydata DraggablePolygon.lock = self def on_motion(self, event): 'on motion we will move the rect if the mouse is over us' if DraggablePolygon.lock is not self: return if event.inaxes != self.poly.axes: return x0, y0, xpress, ypress = self.press dx = event.xdata - xpress dy = event.ydata - ypress xdx = [i+dx for i,_ in geometry] ydy = [i+dy for _,i in geometry] self.newGeometry = [[a, b] for a, b in zip(xdx, ydy)] #polygon = Polygon(self.newGeometry, closed=False, fill=False, linewidth=3, color='#F97306') #patches = [] #patches.append(polygon) #plt.cla() #p = PatchCollection(patches, match_original=True) #ax.add_collection(p) self.poly.set_xy(newGeometry) # this will set the vertices of the polygon self.poly.figure.canvas.draw() def on_release(self, event): 'on release we reset the press data' if DraggablePolygon.lock is not self: return self.press = None DraggablePolygon.lock = None def disconnect(self): 'disconnect all the stored connection ids' self.poly.figure.canvas.mpl_disconnect(self.cidpress) self.poly.figure.canvas.mpl_disconnect(self.cidrelease) self.poly.figure.canvas.mpl_disconnect(self.cidmotion) fig = plt.figure() ax = fig.add_subplot(111) geometry = [[0.0,0.0],[0.1,0.05],[0.2,0.15],[0.3,0.20],[0.4,0.25],[0.5,0.30], [0.6,0.25],[0.7,0.15],[0.8,0.05],[0.9,0.025],[1.0,0.0]] patches = [] polygon = plt.Polygon(geometry, closed=True, fill=False, linewidth=3, color='#F97306') #patches.append(polygon) #p = PatchCollection(patches, match_original=True) #ax.add_collection(p) ax.add_patch(polygon) dp = DraggablePolygon(polygon) dp.connect() plt.show()必须保存在某个地方，但我想应该将其覆盖。

编辑：如以下注释中所建议，我将添加补丁而不是集合，因为我只会绘制单个多边形（请参阅注释掉的旧代码）。另外，我关闭了多边形以演示只能通过在多边形内部单击而不是单击其边缘来拖动补丁。

如果我想再次拖动多边形，它将自动跳回其初始位置。

geometry

我假设newGeometry和def parser_od(url): price=[] url_of = url driver.get(url_of) try: price.append(browser.find_element_by_xpath("//*[@id='root']/article/header/div[2]/div[1]/div[2]").text.replace(" ","").replace("zł","").replace(",",".")) except NoSuchElementException: price.append("")的定义必须在代码内的其他位置，但是经过几次尝试，我找不到有效的解决方案。有人找到我犯的错误吗？

Answer 1

问题下方的注释最终帮助找到了有效的代码。也许这不是最好，最pythonic的方法，但它确实满足了我的要求。

import matplotlib.pyplot as plt
from matplotlib.patches import Polygon

class DraggablePolygon:
    lock = None
    def __init__(self):
        print('__init__')
        self.press = None

        fig = plt.figure()
        ax = fig.add_subplot(111)

        self.geometry = [[0.0,0.0],[0.1,0.05],[0.2,0.15],[0.3,0.20],[0.4,0.25],[0.5,0.30],
                    [0.6,0.25],[0.7,0.15],[0.8,0.05],[0.9,0.025],[1.0,0.0]]
        self.newGeometry = []
        poly = plt.Polygon(self.geometry, closed=True, fill=False, linewidth=3, color='#F97306')
        ax.add_patch(poly)
        self.poly = poly

    def connect(self):
        'connect to all the events we need'
        print('connect')
        self.cidpress = self.poly.figure.canvas.mpl_connect(
        'button_press_event', self.on_press)
        self.cidrelease = self.poly.figure.canvas.mpl_connect(
        'button_release_event', self.on_release)
        self.cidmotion = self.poly.figure.canvas.mpl_connect(
        'motion_notify_event', self.on_motion)

    def on_press(self, event):
        'on button press we will see if the mouse is over us and store some data'
        print('on_press')
        if event.inaxes != self.poly.axes: return
        if DraggablePolygon.lock is not None: return
        contains, attrd = self.poly.contains(event)
        if not contains: return

        if not self.newGeometry:
            x0, y0 = self.geometry[0]
        else:
            x0, y0 = self.newGeometry[0]

        self.press = x0, y0, event.xdata, event.ydata
        DraggablePolygon.lock = self

    def on_motion(self, event):
        'on motion we will move the rect if the mouse is over us'
        if DraggablePolygon.lock is not self:
            return
        if event.inaxes != self.poly.axes: return
        x0, y0, xpress, ypress = self.press
        dx = event.xdata - xpress
        dy = event.ydata - ypress

        xdx = [i+dx for i,_ in self.geometry]
        ydy = [i+dy for _,i in self.geometry]
        self.newGeometry = [[a, b] for a, b in zip(xdx, ydy)]
        self.poly.set_xy(self.newGeometry)
        self.poly.figure.canvas.draw()

    def on_release(self, event):
        'on release we reset the press data'
        print('on_release')
        if DraggablePolygon.lock is not self:
            return

        self.press = None
        DraggablePolygon.lock = None
        self.geometry = self.newGeometry


    def disconnect(self):
        'disconnect all the stored connection ids'
        print('disconnect')
        self.poly.figure.canvas.mpl_disconnect(self.cidpress)
        self.poly.figure.canvas.mpl_disconnect(self.cidrelease)
        self.poly.figure.canvas.mpl_disconnect(self.cidmotion)


dp = DraggablePolygon()
dp.connect()

plt.show()

如何绘制可拖动的多边形

1 个答案: