Plotting with 3D axes

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

How to add 3D axes to a figure

We will use the Matplotlib inline magic in Jupyter Notebook to create our plotting setup. Begin by making a new figure and add some axes to this figure. Add a subplot, as shown in the following code, along with the projection= '3D' keyword argument:

# Make a Axes3D
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d...