In [ ]:

    

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


    

In [ ]:

    

fig = plt.figure()
ax = fig.gca(projection='3d')
x = np.arange(-5,5,0.25)
y = np.arange(-5,5,0.25)

x, y = np.meshgrid(x,y)
r = np.sqrt(x**2 + y**2)
z = np.sin(r)

surf = ax.plot_surface(x,y,z,rstride=1,cstride=1,cmap=cm.coolwarm, 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()


    

In [ ]: