13.6 Moving to three dimensions

matplotlib can plot in three dimensions if we import the tools we need from mpl_toolkits.mplot3d. To help you see the main components of a 3D plot, here is an example where I have drawn the point at the origin (0, 0, 0) and specified the axes ranges and labels.

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# --------------------------------------------------------------
# create the plot figure
figure = plt.figure()

# create a 3D axes
axes = plt.axes(projection='3d')
# --------------------------------------------------------------

# label the x, y, and z axes in a large font
axes.set_xlabel('x', fontsize="16")
axes.set_ylabel('y', fontsize="16")
axes.set_zlabel('z', fontsize="16")

# set the limits of the values shown on the axes
axes.set_xlim(0, 1)
axes.set_ylim(0, 1)
axes.set_zlim(0, 1)

# label the origin
axes.text(0...