from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure(1)

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

#2d Scaler field plot ax.plot_wireframe = wire frame plot

#2d ax.plot_surface = color the surface

x = y = np.linspace(-10., 10., 41)

xv, yv = np.meshgrid(x, y, indexing='ij', sparse=False)

hv = h0/(1 + (xv**2+yv**2)/(R**2))

fig = plt.figure(1)

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

ax.plot_wireframe(xv,yv,hv,rstride = 1 , cstride = 4)