We will begin by importing numpy, matplotlib, and pyplot, as shown here:
import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
%matplotlib notebook
# Set up figure size and DPI for screen demo
plt.rcParams['figure.figsize'] = (6,4)
plt.rcParams['figure.dpi'] = 150