# -*- coding: utf-8 -*-importnumpyasnpfrommatplotlib.figureimportFigurefrommatplotlib.textimportTextimportmatplotlib# noqafrommatplotlib.pyplotimportfigurefrommatplotlib.widgetsimportButtonfrommpl_toolkits.axes_grid1.axes_sizeimportFixedfrommpl_toolkits.axes_grid1.axes_dividerimportmake_axes_locatable# matplotlib.use('Agg', force=True)classFigure1(Figure):"""Interactive figure with scrollbars, for data browsing."""def__init__(self,yticks,**kwargs):super().__init__(figsize=(6,6))ax_main=self.add_subplot(1,1,1,position=[0.1,0.1,0.9,0.9])div=make_axes_locatable(ax_main)ax_button=div.append_axes(position='bottom',size=Fixed(0.5),pad=0)self._button=Button(ax_button,'Help')# main plotoffsets=np.arange(len(yticks))ax_main.set(xlim=[0,1],yticks=offsets)ax_main.set_yticklabels(yticks,picker=True)ax_main.plot([0,1], [offsets,offsets])picked=list()defon_pick(event):"""Handle matplotlib pick events."""ifisinstance(event.artist,Text):ch_name=event.artist.get_text()print(f'pick{ch_name}')picked.append(ch_name)yticks=list(x*5forxin'ABCD')fig=figure(FigureClass=Figure1,yticks=yticks)fig.canvas.mpl_connect('pick_event',on_pick)fig.canvas.draw()ax=fig.axes[0]x,y=ax.transData.transform_point((-0.1,0))func=fig.canvas.button_press_event(x=x,y=y,button=1,guiEvent=None)assert'AAAAA'inpickedassertlen(picked)==1,len(picked)