在matplotlib中动态更新图形

时间:2013-08-21 13:55:09

标签: python matplotlib tkinter

# when the code works but it's meaningless to include it
### When I can't get this part to work and I'd need your code

如何在matplotlib中隐藏或显示Axes对象(子图),以便在同一图中的不同轴之间切换?

我正在使用matplotlib在Tkinter GUI中显示图形,我想使用radiobuttons在同一图中的不同轴之间切换。

基本上我会将一些radiobuttons链接到IntVar():

graphic_version = tk.IntVar()
tk.Radiobutton(root, text='Option1', variable=graphic_version, value=1).pack()
tk.Radiobutton(root, text='Option2', variable=graphic_version, value=2).pack()

然后我会使用自定义方法跟踪IntVar(),并使用所请求的图形更新我的数字:

choice.trace("w", lambda choice: myGraphic.showGraphic(version))

这样每当用户点击一个单选按钮时,该图就会使用不同版本的图更新。现在问题是我不知道如何正确地做showGraphic。假设我使用这个类系统来获得绘制相同数据的2个不同版本:

class Version1():
    def __init__(self, ax, data):
        self.ax = ax #This is a Axes object
        self.data = self._formatDataV1(data)

        self._draw()
        self._setOptions()
        self.hide()

    def _formatDataV1(self, data):
        #Here I manipulate the raw data to extract the info I need for this version
        #Just a bunch of math algorithms it works fine

    def _setOptions(self):
        #Here I can overwrite or change settings specific for this version

    def _draw(self):
        self.ax.bar(self.data[0], self.data[1], width=1, color='red')
        self._setOptions()

    def hide(self):
        ###How do I remove the ax without affecting the figure?

    def show(self):
        ###If I want to see this version again I don't want the cost of redrawing   


class Version2():
    def __init__(self, ax, data):
        self.ax = ax #This is a Axes object
        self.data = self._formatDataV1(data)

        self._draw()
        self._setOptions()
        self.hide()

    def _formatDataV2(self, data):
        #The data is manipulated differently here to extract new information

    def _setOptions(self):
        #These options are specific to the version2 to get the display right

    def _draw(self): #Drawing a different version of the graphic with differently formated data
        self.ax.plot(self.data[0], self.data[1])
        self._setOptions()

    def hide(self):
        ###How do I remove the ax without affecting the figure?

    def show(self):
        ###If I want to see this version again I don't want the cost of redrawing

class MyGraphic(tk.LabelFrame):
    def __init__(self, root, data, **options):
        #I use the labelframe only as a container to make things pretty
        tk.LabelFrame.__init__(self, root, text="My 1337 graphic : ", **options)

        self.data = data
        self.fig = mpl.figure.Figure()
        self.ax = self.fig.add_subplot(111)

        self._drawCanvas() #This is just for Tkinter compatibility

        self.my_versions = {}
        self.my_versions.update({'v1' : Version1(self.ax, self.data)})
        self.my_versions.update({'v2' : Version2(self.ax, self.data)})

    def _drawCanvas(self):
        self.canvas = FigureCanvasTkAgg(self.figure, master=self)
        self.canvas.show()
        self.canvas.get_tk_widget().grid()
        self.canvas._tkcanvas.pack(side=tk.TOP, fill=tk.BOTH, expand=1)

    def _setOptions(self, **kwargs):
        #Here I can set options common to both versions of the graphic

    def showGraphic(self, graphic_version):
        for i, item in enumerate(self.my_versions):
            item.hide()
        if graphic_version == 1:
            self.my_versions['v1'].show()
        elif graphic_version == 2:
            self.my_versions['v2'].show()

        self._setOptions()

对于冗长的帖子感到抱歉,但我宁愿包含太多细节,并编辑那些在解决后不必要的人。

基本上我希望能够隐藏并在同一个数字上显示不同的斧头,具体取决于我的用户所做的选择。这个难题的缺失部分是myGraphic.show()和myGraphic.hide()。

我也是一个完整的matplotlib newby我尝试了这个设计,因为在需要时它似乎很容易实现其他版本,但设计输入也非常受欢迎。

2 个答案:

答案 0 :(得分:0)

您可以使用figure.delaxes从图中删除轴(并使用add_axes添加): http://matplotlib.org/api/figure_api.html#matplotlib.figure.Figure.delaxes

答案 1 :(得分:0)

我设法通过figure.clear()figure.add_axes(ax)来解决问题 我将尝试编辑一个干净的问答 - 明天我有一个最小的例子,如何在Tkinter的同一个数字上切换不同版本的情节。