Figura 3d
Preparar datos
X, Y = np.meshgrid(np.linspace(-1, 1, 100), np.linspace(-1, 1, 100))
D = np.sqrt(X**2 + Y**2)
sigmu, mu = 1.0, 0.0
G = np.exp(-(D-mu)**2 / (2.0*sigma**2))
Mapa de contorno
plt.contour(X, Y, G)
plt.show()
Dibujo 3D
Versión ingenua
fig = plt.figure()
# gca: get current axis
# projection='3d': use Cartesian coordinates
ax = fig.gca(projection='3d')
ax.plot_surface(X, Y, G)
plt.show()
Versión de color
# cm: color mapping
from matplotlib import cm
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.plot_surface(X, Y, G, cmap=cm.coolwarm)
plt.show()