#Exemplo: plotagem de superfície em 3 dimensões
#vamos plotar o gráfico de z = x^2 + 0.5y^3 para x de -5 até 5
# e y de -5 até 5
from mpl_toolkits.mplot3d import Axes3D   #importa os eixos em 3D
from matplotlib import cm                   #importa mapa de cores
import matplotlib.pyplot as plt
import numpy as np

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

x = np.arange(-5, 5, 0.25)
y = np.arange(-5, 5, 0.25)

#gerando X e Y como matrizes que representam o produto cartesiano x por y
X, Y  = np.meshgrid(x, y)
Z = X**2 + 0.5*Y**3

surf = ax.plot_surface(X, Y, Z, rstride = 1, cstride = 1, cmap=cm.jet,
                       linewidth=10)
fig.colorbar(surf)

plt.show()