We will plot in three dimensions a simple three-dimensional function:
3d
keyword to specify a three-dimensional projection for the plot.ax = fig.add_subplot(111, projection=’3d’)
meshgrid
function. This will be used to initialize the x
and y
values.u = np.linspace(-1, 1, 100) x, y = np.meshgrid(u, u)
ax.plot_surface(x, y, z, rstride=4, cstride=4,cmap=cm.YlGnBu_r)
The result is the following 3D plot:
We created a plot of a three dimensional function (see three_d.py
):
from mpl_toolkits.mplot3d import Axes3D import matplotlib.pyplot as plt import numpy as np from matplotlib import cm fig = plt.figure() ax = fig.add_subplot(111, projection=’3d’) u = np.linspace(-1, 1, 100) x, y = np.meshgrid(u, u) z = x ** 2 + y ** 2 ax.plot_surface(x, y, z, rstride=4, cstride=4, cmap=cm.YlGnBu_r) plt.show()