在wxPython位图中显示matplotlib图

import wx
import matplotlib.pyplot as plt
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigureCanvas
from matplotlib.figure import Figure

class MyFrame(wx.Frame):
    def __init__(self, parent, title):
        super(MyFrame, self).__init__(parent, title=title, size=(800, 600))
        self.InitUI()

    def InitUI(self):
        # 创建一个matplotlib图形对象
        fig = Figure()
        # 在图形对象上创建一个子图
        self.ax = fig.add_subplot(111)
        # 绘制matplotlib图形
        self.ax.plot([1, 2, 3, 4], [1, 4, 9, 16])

        # 创建一个wxPython位图对象
        self.bitmap = wx.Bitmap(800, 600)
        # 创建一个wxPython位图画布对象
        self.canvas = FigureCanvas(self, -1, fig)
        # 将位图画布对象绑定到位图对象上
        self.canvas.SetBitmap(self.bitmap)

        # 将位图画布对象添加到窗口布局中
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.canvas, 1, wx.EXPAND)
        self.SetSizer(sizer)

app = wx.App()
frame = MyFrame(None, "wxPython with Matplotlib")
frame.Show()
app.MainLoop()