Matplotlib from Part 03

3D surface (color map)

Demonstrates plotting a 3D surface colored with the coolwarm color map.
The surface is made opaque by using antialiased=False.

Also demonstrates using the LinearLocator and custom formatting for the
z axis tick labels.

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import numpy as np

plt.rcParams['axes.facecolor'] = '#ffff97ff'

fig = plt.figure()
ax = fig.gca(projection='3d')


# Make data.
X = np.arange(-2, 2, 0.02)
Y = np.arange(-2, 2, 0.02)
X, Y = np.meshgrid(X, Y)
Z = X**2*Y/(X**4+Y*Y) # This is is the function from example (c)

# Plot the surface.
surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm,
                       linewidth=0, antialiased=False)

# Customize the z axis.
ax.set_zlim(-2.01, 2.01)

# Add a color bar which maps values to colors.
fig.colorbar(surf, shrink=0.5, aspect=5)