Python drawing basics

3d figure

Prepare data

X, Y = np.meshgrid(np.linspace(-1, 1, 100), np.linspace(-1, 1, 100)) 
D = np.sqrt(X**2 + Y**2)
sigmu, mu = 1.0, 0.0
G = np.exp(-(D-mu)**2 / (2.0*sigma**2))

Contour map

plt.contour(X, Y, G)
plt.show()

3D drawing

Naive version

fig = plt.figure() 
# gca: get current axis
# projection='3d': use Cartesian coordinates
ax = fig.gca(projection='3d')
ax.plot_surface(X, Y, G)
plt.show()

Color version

# cm: color mapping
from matplotlib import cm
fig = plt.figure() 
ax = fig.gca(projection='3d')
ax.plot_surface(X, Y, G, cmap=cm.coolwarm)
plt.show()