the code in jupyter notebook

matplot_demo