%matplotlib notebook %load_ext autoreload %autoreload 2 import numpy as np import matplotlib.pyplot as plt