from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np
fig = plt.figure()
ax = fig.gca(projection='3d')
Xv = np.arange(-5, 5, 0.05)
Yv = np.arange(-5, 5, 0.05)
Xv, Yv = np.meshgrid(Xv, Yv)
R = np.sqrt(Xv**2 + Yv**2)
Zv = np.sin(R)
surf = ax.plot_surface(Xv, Yv, Zv, rstride=1, cstride=1, cmap=plt.cm.jet,
linewidth=0, antialiased=False)
ax.set_zlim(-1.01, 1.01)
ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))
fig.colorbar(surf, shrink=0.5, aspect=5)
plt.show()