#!/usr/bin/env python
import matplotlib
import matplotlib.pyplot as plt
matplotlib.use('WXAgg')
import numpy as np
import wx

class TestFrame(wx.Frame):
    def __init__(self, parent, title):
        super(TestFrame, self).__init__(parent, title=title,size=(350, 250))
        self.sz=wx.BoxSizer()
        self.button=wx.Button(self,-1,"click")
        self.button.Bind(wx.EVT_BUTTON, self.onClick)   
        self.sz.Add(self.button,wx.EXPAND)
        self.SetSizer(self.sz)
        
    def onClick(self,evt):
        plt.plot(np.arange(0,6.28,0.1),np.sin(np.arange(0,6.28,0.1)))
        plt.show(block=True)
        ## as wx.App mainloop is running, matplotlib won't create an event loop. 
        ## we therefore need to manually create one!
        #plt.gcf().canvas.start_event_loop() #!magick line
        print("This shouldn't get printed before the closing the pyplot window!")

def main():
    app = wx.App()
    tf = TestFrame(None, title='Test app')
    tf.Show()
    app.MainLoop()

if __name__ == '__main__':
    main()