# L-16 MCS 507 Wed 28 Sep 2011 : cubic_cylinder.py

# This script uses pyplot of matplotlib to plot
# the cubic cylinder z = x^3.

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
import matplotlib.pyplot as plt
import numpy as np

f = plt.figure()
a = f.gca(projection='3d',azim=-117,elev=15)
x = np.arange(-1,1,0.03)
y = np.arange(-1,1,0.03)
X, Y = np.meshgrid(x,y)
Z = X**3
for i in xrange(X.shape[0]):
  for j in xrange(X.shape[1]):
     if X[i,j]**2 < Y[i,j]: Z[i,j] = -1

s = a.plot_surface(X,Y,Z,rstride=1,cstride=1,
                   cmap=cm.jet,linewidth=0)
f.colorbar(s,shrink=0.5)

plt.show()