matplotlib鼠标事件绘制圆.docVIP

#coding=utf8

importmatplotlib.pyplotasplt

#importmatplotlib.animationasanimation

importnumpyasnp

importmatplotlib.linesaslines

fromcreat_polygonimportcreatpolygon

frommatplotlib.widgetsimportCursor

frommatplotlib.patchesimportEllipse,Rectangle

frommatplotlibimporttext

frommatplotlib.backends.backend_qt5aggimportFigureCanvasQTAggasFigureCanvas

frommatplotlib.figureimportFigure

classonclick_draw_circl():

鼠标绘制椭圆、圆形，鼠标点击按下处为中心，移动鼠标椭圆变化，释放鼠标完成绘图，该为绘图类，应用是需输入fig窗口aritst。如果同时按下shift则绘制圆

def__init__(self,myfig):

self.fig=myfig

self.ax=plt.gca()

self.connectall()

self.onpress=False

self.cl=[]

print(myfig)

defconnectall(self):

self.cid1=self.fig.canvas.mpl_connect(button_press_event,self.btpress)

self.cid2=self.fig.canvas.mpl_connect(motion_notify_event,self.btmotion)

self.cid3=self.fig.canvas.mpl_connect(button_release_event,self.btrelease)

defbtpress(self,event):

self.x0=event.xdata

self.y0=event.ydata

self.onpress=True

print(self.x0,self.y0)

defbtmotion(self,event):

ifself.onpress:

self.x=event.xdata

self.y=event.ydata

dltx=abs(self.x-self.x0)*2

dlty=abs(self.y-self.y0)*2

ifevent.key==shift:#若果同时按下shift键则绘制圆，dlty=dltx

dlty=dltx

iflen(self.cl)!=0:

temph=self.cl.pop()

temph.remove()

#self.hands.remove()

self.hands=Ellipse(xy=(self.x0,self.y0),width=dltx,height=dlty,fill=False,alpha=0.9,picker=5)

self.hands.set(linewidth=1.5,edgecolor=r)

self.ax.add_patch(self.hands)

self.cl.append(self.hands)

plt.gca().figure.canvas.draw()#重绘画布图像

self.cl.append(self.hands)

defbtrelease(self,event):

self.onpress=False

self.cl[:]=[