We will plot a simple three-dimensional function:
ax = fig.add_subplot(111, projection='3d')
meshgrid()
function 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 three-dimensional 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()