1 from numpy import *
   2 import pylab as p
   3 #import matplotlib.axes3d as p3
   4 import mpl_toolkits.mplot3d.axes3d as p3
   5 
   6 # u and v are parametric variables.
   7 # u is an array from 0 to 2*pi, with 100 elements
   8 u=r_[0:2*pi:100j]
   9 # v is an array from 0 to 2*pi, with 100 elements
  10 v=r_[0:pi:100j]
  11 # x, y, and z are the coordinates of the points for plotting
  12 # each is arranged in a 100x100 array
  13 x=10*outer(cos(u),sin(v))
  14 y=10*outer(sin(u),sin(v))
  15 z=10*outer(ones(size(u)),cos(v))

   1 fig=p.figure()
   2 ax = p3.Axes3D(fig)
   3 ax.plot_wireframe(x,y,z)
   4 ax.set_xlabel('X')
   5 ax.set_ylabel('Y')
   6 ax.set_zlabel('Z')
   7 p.show()

   1 # this connects each of the points with lines
   2 fig=p.figure()
   3 ax = p3.Axes3D(fig)
   4 # plot3D requires a 1D array for x, y, and z
   5 # ravel() converts the 100x100 array into a 1x10000 array
   6 ax.plot3D(ravel(x),ravel(y),ravel(z))
   7 ax.set_xlabel('X')
   8 ax.set_ylabel('Y')
   9 ax.set_zlabel('Z')
  10 fig.add_axes(ax)
  11 p.show()

   1 fig=p.figure()
   2 ax = p3.Axes3D(fig)
   3 # scatter3D requires a 1D array for x, y, and z
   4 # ravel() converts the 100x100 array into a 1x10000 array
   5 ax.scatter3D(ravel(x),ravel(y),ravel(z))
   6 ax.set_xlabel('X')
   7 ax.set_ylabel('Y')
   8 ax.set_zlabel('Z')
   9 p.show()

   1 fig=p.figure()
   2 ax = p3.Axes3D(fig)
   3 # x, y, and z are 100x100 arrays
   4 ax.plot_surface(x,y,z)
   5 ax.set_xlabel('X')
   6 ax.set_ylabel('Y')
   7 ax.set_zlabel('Z')
   8 p.show()

   1 delta = 0.025
   2 x = arange(-3.0, 3.0, delta)
   3 y = arange(-2.0, 2.0, delta)
   4 X, Y = p.meshgrid(x, y)
   5 Z1 = p.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
   6 Z2 = p.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
   7 # difference of Gaussians
   8 Z = 10.0 * (Z2 - Z1)
   9 fig=p.figure()
  10 ax = p3.Axes3D(fig)
  11 ax.contour3D(X,Y,Z)
  12 ax.set_xlabel('X')
  13 ax.set_ylabel('Y')
  14 ax.set_zlabel('Z')
  15 p.show()

   1 # in mplt3D change:
   2 # levels, colls = self.contourf(X, Y, Z, 20)
   3 # to:
   4 # C = self.contourf(X, Y, Z, *args, **kwargs)
   5 # levels, colls = (C.levels, C.collections)
   6 fig=p.figure()
   7 ax = p3.Axes3D(fig)
   8 ax.contourf3D(X,Y,Z)
   9 ax.set_xlabel('X')
  10 ax.set_ylabel('Y')
  11 ax.set_zlabel('Z')
  12 fig.add_axes(ax)
  13 p.show()

   1 x=r_[-10:10:100j]
   2 y=r_[-10:10:100j]
   3 z= add.outer(x*x, y*y)
   4 ### Contour plot of z = x**2 + y**2
   5 p.contour(x,y,z)
   6 ### ContourF plot of z = x**2 + y**2
   7 p.figure()
   8 p.contourf(x,y,z)
   9 p.show()