10 import matplotlib.pyplot
as plt
11 import matplotlib.cm
as cm
16 if (pv.shape != psi.shape):
17 raise "pv.shape != psi.shape" 30 ax = plt.subplot(2,1,1)
31 ax.set_autoscale_on(
False)
32 plt.axis([0, nx+1, 0, ny+1])
33 plt.title(
"Upper Layer")
35 CSPV = plt.contourf(x,y, pv[0,:,:],20,cmap=CM)
36 CSPSI = plt.contour (x,y,psi[0,:,:],10,colors=
'k')
38 plt.clabel(CSPSI, fontsize=9, inline=1)
39 CB = plt.colorbar(CSPV)
42 bx = plt.subplot(2,1,2)
43 bx.set_autoscale_on(
False)
44 plt.axis([0, nx+1, 0, ny+1])
45 plt.title(
"Lower Layer")
47 CSPV = plt.contourf(x,y, pv[1,:,:],20,cmap=CM)
48 CSPSI = plt.contour (x,y,psi[1,:,:],10,colors=
'k')
50 plt.clabel(CSPSI, fontsize=9, inline=1)
51 CB = plt.colorbar(CSPV)