wxpython面板中的matplotlib动画

Question

我在结合 wxPython 和 matplotlib 的应用程序上苦苦挣扎。

我想在 wxPanel 中嵌入动画 matplotlib object。数据应在运行时添加。

我的模块代码：

（我无法获得正确的格式，请参阅 http://pastebin.com/PU5QFEzG）

'''
a panel to display a given set of data in a wxframe as a heatmap, using pcolor
from the matplotlib

@author: me
'''
import wx
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigureCanvas #todo: OW 26.10.15 needed?

class plotPanel(wx.Panel):
    def __init__(self, parent):
        wx.Panel.__init__(self, parent)
        self.figure = plt.Figure()
        self.subplot = self.figure.add_subplot(111)
        plt.title('test')
        self.canvas = FigureCanvas(self, -1, self.figure)  #ToDo: OW 26.10.15 Verstehen
        self.sizer = wx.BoxSizer(wx.VERTICAL)
        self.sizer.Add(self.canvas, 1, wx.LEFT | wx.TOP | wx.GROW)
        self.SetSizer(self.sizer)
        self.Fit()
        self.dataSet = []
        self.animator = animation.FuncAnimation(self.figure,self.anim, interval=1000)

    def anim(self, a):
        if(len(self.dataSet) == 0):
            return 0
        i = a % len(self.dataSet)
        obj = self.subplot.pcolor(self.dataSet[i], cmap='RdBu')
        return obj

    def add_data(self, data):
        self.dataSet.append(data)


#
#    Code for a standalone test run
#
class TestFrame(wx.Frame):
    def __init__(self,parent,title):
        wx.Frame.__init__(self,parent,title=title,size=(1000,1000))
        self.statusbar = self.CreateStatusBar()
        self.statusbar.SetStatusText('Status Bar')


if __name__ == '__main__':
    from numpy.random import rand #todo: OW 26.10.15 remove
    app = wx.App(redirect=False)
    frame = TestFrame(None, 'Debug Frame')
    panel = plotPanel(frame)
    frame.Show()
    C = rand(10,10)
    panel.add_data(C)
    C = rand(10,10)
    panel.add_data(C)
    C = rand(10,10)
    panel.add_data(C)
    app.MainLoop()

我现在正在努力向图表添加更多详细信息，例如颜色栏或标题。

如果我在 anim_Fkt 中添加 self.subplot.title = 'test'，我会得到“'str' object 没有属性 'get_animated'”。如果我尝试 plt.title('test')，它没有任何效果。添加标题或颜色栏或图例的正确方法是什么？

Answer 1

要将特征添加到嵌入式 matplotlib 图形，您必须使用 object-oriented matplotlib API。一个 matplotlib 图由一个图形（一个 matplotlib.figure.Figure 实例）组成，其中有一个或多个轴（matplotlib.axes.Axes 个实例）。他们每个人依次包含 Drawables（线条、图像、散点图等）。

可以使用 setter 方法将标题添加（或修改）到 Figure 或每个 Axes，例如 Figure.suptitle 或 Axes.set_title （或者，在您的情况下，self.figure.suptitle() 或 self.subplot.set_title）。

颜色条有点棘手，因为它需要创建数据 object（可映射）：我们只能在 anim 函数中创建它。另外，我们不想创建很多 colorbar 个实例；一旦创建，我们只需要更新它。实现起来很简单：在构造函数中用 None 实例化 self.colorbar，然后在动画函数中对照 None 进行检查：如果是 None，则创建颜色条，如果是没有，那就更新吧：

class plotPanel(wx.Panel):
    def __init__(self, parent):
        ...
        self.colorbar = None

    def anim(self, a):
        ...
        self.subplot.set_title("Frame %i" % a)
        if self.colorbar is None:
            self.colorbar = self.figure.colorbar(obj)
        else:
            self.colorbar.update_normal(obj)
        self.colorbar.update_normal(obj)
        return obj

wxpython面板中的matplotlib动画

matplotlib animation in wxpython panel

python

wxwidgets

wxpython

matplotlib