Refactor display helper function

mkcor · grlee77 · mkcor · commit dda908cfb7e1 · 2020-12-29T13:30:52.000+01:00
Co-authored-by: Gregory R. Lee &lt;grlee77@gmail.com&gt;
diff --git a/doc/examples/applications/plot_3d_image_processing.py b/doc/examples/applications/plot_3d_image_processing.py
@@ -108,20 +108,15 @@ def show_plane(ax, plane, cmap="gray", title=None):
 
 #####################################################################
 # As hinted before, a three-dimensional image can be viewed as a series of
-# two-dimensional planes. Let us write a helper function, `display`, to
-# display 30 planes of our data. By default, every other plane is displayed.
+# two-dimensional planes. Let us write a helper function, `display`, to create
+# a montage of several planes. By default, every other plane is displayed.
 
 
-def display(im3d, cmap="gray", step=2):
-    _, axes = plt.subplots(nrows=5, ncols=6, figsize=(16, 14))
-
-    vmin = im3d.min()
-    vmax = im3d.max()
-
-    for ax, image in zip(axes.flatten(), im3d[::step]):
-        ax.imshow(image, cmap=cmap, vmin=vmin, vmax=vmax)
-        ax.set_xticks([])
-        ax.set_yticks([])
+def display(im3d, cmap='gray', step=2):
+    data_montage = util.montage(im3d[::step], padding_width=4, fill=np.nan)
+    _, ax = plt.subplots(figsize=(16, 14))
+    ax.imshow(data_montage, cmap=cmap)
+    ax.set_axis_off()
 
 
 display(data)
