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

import numpy as np


class CanvasFrame(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self, None, -1,
                          'CanvasFrame', size=(550, 350))

        self.SetBackgroundColour('blue')

        ###############重要部分S###############
        self.figure = Figure()
        self.axes = self.figure.add_subplot(111)

        x = np.linspace(-4, 4, 30)
        y = np.sin(x)

        self.axes.plot(x, y, '--*b')

        self.canvas = FigureCanvas(self, -1, self.figure)
        ###############重要部分E###############

        self.sizer = wx.BoxSizer(wx.VERTICAL)
        self.sizer.Add(self.canvas, 1, wx.LEFT | wx.TOP | wx.GROW)
        self.SetSizer(self.sizer)
        self.Fit()


class MyApp(wx.App):
    def OnInit(self):
        'Create the main window and insert the custom frame'
        frame = CanvasFrame()
        frame.Show(True)

        return True


app = MyApp()
app.MainLoop()