.. _mplot3d-surface3d_radial_demo:

mplot3d example code: surface3d_radial_demo.py
==============================================



.. plot:: /media/TOSHI/temp/numpy_scipy_matplotlib/matplotlib/matplotlib-1.1.1~rc2/doc/mpl_examples/mplot3d/surface3d_radial_demo.py

::

    # By Armin Moser
    
    from mpl_toolkits.mplot3d import Axes3D
    import matplotlib
    import numpy as np
    from matplotlib import cm
    from matplotlib import pyplot as plt
    step = 0.04
    maxval = 1.0
    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')
    
    # create supporting points in polar coordinates
    r = np.linspace(0,1.25,50)
    p = np.linspace(0,2*np.pi,50)
    R,P = np.meshgrid(r,p)
    # transform them to cartesian system
    X,Y = R*np.cos(P),R*np.sin(P)
    
    Z = ((R**2 - 1)**2)
    ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.jet)
    ax.set_zlim3d(0, 1)
    ax.set_xlabel(r'$\phi_\mathrm{real}$')
    ax.set_ylabel(r'$\phi_\mathrm{im}$')
    ax.set_zlabel(r'$V(\phi)$')
    plt.show()
    

Keywords: python, matplotlib, pylab, example, codex (see :ref:`how-to-search-examples`)