STY: update with use of plt.subplots(), other readability edits

mscross · mscross · commit eed437abe17b · 2015-04-20T13:30:27.000-05:00
diff --git a/examples/images_contours_and_fields/pcolormesh_levels.py b/examples/images_contours_and_fields/pcolormesh_levels.py
@@ -18,7 +18,7 @@
 y, x = np.mgrid[slice(1, 5 + dy, dy),
                 slice(1, 5 + dx, dx)]
 
-z = np.sin(x) ** 10 + np.cos(10 + y * x) * np.cos(x)
+z = np.sin(x)**10 + np.cos(10 + y*x) * np.cos(x)
 
 # x and y are bounds, so z should be the value *inside* those bounds.
 # Therefore, remove the last value from the z array.
@@ -31,22 +31,23 @@
 cmap = plt.get_cmap('PiYG')
 norm = BoundaryNorm(levels, ncolors=cmap.N, clip=True)
 
-plt.subplot(2, 1, 1)
-im = plt.pcolormesh(x, y, z, cmap=cmap, norm=norm)
-plt.colorbar()
-# set the limits of the plot to the limits of the data
-plt.axis([x.min(), x.max(), y.min(), y.max()])
-plt.title('pcolormesh with levels')
+fig, (ax0, ax1) = plt.subplots(nrows=2)
+
+im = ax0.pcolormesh(x, y, z, cmap=cmap, norm=norm)
+fig.colorbar(im, ax=ax0)
+ax0.set_title('pcolormesh with levels')
 
 
-plt.subplot(2, 1, 2)
 # contours are *point* based plots, so convert our bound into point
 # centers
-plt.contourf(x[:-1, :-1] + dx / 2.,
-             y[:-1, :-1] + dy / 2., z, levels=levels,
-             cmap=cmap)
-plt.colorbar()
-plt.title('contourf with levels')
-
+cf = ax1.contourf(x[:-1, :-1] + dx/2.,
+                  y[:-1, :-1] + dy/2., z, levels=levels,
+                  cmap=cmap)
+fig.colorbar(cf, ax=ax1)
+ax1.set_title('contourf with levels')
+
+# adjust spacing between subplots so `ax1` title and `ax0` tick labels
+# don't overlap
+fig.tight_layout()
 
 plt.show()
