added demonstration of separate horizontal/vertical axes padding

matejak · matejak · commit 92f5e914b339 · 2014-02-27T00:24:06.000+01:00
diff --git a/examples/axes_grid/demo_axes_grid.py b/examples/axes_grid/demo_axes_grid.py
@@ -14,7 +14,7 @@ def demo_simple_grid(fig):
     A grid of 2x2 images with 0.05 inch pad between images and only
     the lower-left axes is labeled.
     """
-    grid = AxesGrid(fig, 131, # similar to subplot(131)
+    grid = AxesGrid(fig, 141, # similar to subplot(141)
                     nrows_ncols = (2, 2),
                     axes_pad = 0.05,
                     label_mode = "1",
@@ -33,7 +33,7 @@ def demo_grid_with_single_cbar(fig):
     """
     A grid of 2x2 images with a single colorbar
     """
-    grid = AxesGrid(fig, 132, # similar to subplot(132)
+    grid = AxesGrid(fig, 142, # similar to subplot(142)
                     nrows_ncols = (2, 2),
                     axes_pad = 0.0,
                     share_all=True,
@@ -61,7 +61,7 @@ def demo_grid_with_each_cbar(fig):
     A grid of 2x2 images. Each image has its own colorbar.
     """
 
-    grid = AxesGrid(F, 133, # similar to subplot(122)
+    grid = AxesGrid(F, 143, # similar to subplot(143)
                     nrows_ncols = (2, 2),
                     axes_pad = 0.1,
                     label_mode = "1",
@@ -83,16 +83,47 @@ def demo_grid_with_each_cbar(fig):
     grid.axes_llc.set_xticks([-2, 0, 2])
     grid.axes_llc.set_yticks([-2, 0, 2])
 
+def demo_grid_with_each_cbar_labelled(fig):
+    """
+    A grid of 2x2 images. Each image has its own colorbar.
+    """
+
+    grid = AxesGrid(F, 144, # similar to subplot(144)
+                    nrows_ncols = (2, 2),
+                    axes_pad = ( 0.45, 0.15),
+                    label_mode = "1",
+                    share_all = True,
+                    cbar_location="right",
+                    cbar_mode="each",
+                    cbar_size="7%",
+                    cbar_pad="2%",
+                    )
+    Z, extent = get_demo_image()
+
+    # Use a different colorbar range every time 
+    limits = ((0, 1), (-2, 2), (-1.7, 1.4), (-1.5, 1))
+    for i in range(4):
+        im = grid[i].imshow(Z, extent=extent, interpolation="nearest", 
+                vmin = limits[i][0], vmax = limits[i][1])
+        grid.cbar_axes[i].colorbar(im)
+
+    for i, cax in enumerate(grid.cbar_axes):
+       cax.set_yticks((limits[i][0], limits[i][1]))
+
+    # This affects all axes because we set share_all = True.
+    grid.axes_llc.set_xticks([-2, 0, 2])
+    grid.axes_llc.set_yticks([-2, 0, 2])
 
 
 if 1:
-    F = plt.figure(1, (5.5, 2.5))
+    F = plt.figure(1, (10.5, 2.5))
 
-    F.subplots_adjust(left=0.05, right=0.98)
+    F.subplots_adjust(left=0.05, right=0.95)
 
     demo_simple_grid(F)
     demo_grid_with_single_cbar(F)
     demo_grid_with_each_cbar(F)
+    demo_grid_with_each_cbar_labelled(F)
 
     plt.draw()
     plt.show()
