Matplotlib axes are two-dimensional by default. In order to create three-dimensional plots, we need to import the Axes3D class from the mplot3d toolkit, that will enable a new kind of projection for an axes, namely '3d':
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fi...