added solution from gallery/colormap reference

Ptrskay3 · Ptrskay3 · commit f9b993051504 · 2021-01-22T09:42:45.000+01:00
diff --git a/tutorials/colors/colormaps.py b/tutorials/colors/colormaps.py
@@ -204,19 +204,18 @@
 
 
 def plot_color_gradients(cmap_category, cmap_list):
+    # Create figure and adjust figure height to number of colormaps
     nrows = len(cmap_list)
-    fig, axs = plt.subplots(nrows=nrows + 1, figsize=(6, 0.29 * nrows))
-    fig.subplots_adjust(top=1, bottom=0, left=0.2, right=0.99)
-    axs[0].text(0.5, 0.5, cmap_category + " colormaps", fontsize=14,
-                transform=axs[0].transAxes, horizontalalignment="center",
-                verticalalignment="center")
+    figh = 0.35 + 0.15 + (nrows + (nrows - 1) * 0.1) * 0.22
+    fig, axs = plt.subplots(nrows=nrows + 1, figsize=(6.4, figh))
+    fig.subplots_adjust(top=1 - 0.35 / figh, bottom=0.15 / figh,
+                        left=0.2, right=0.99)
+    axs[0].set_title(cmap_category + ' colormaps', fontsize=14)
 
-    for ax, name in zip(axs[1:], cmap_list):
+    for ax, name in zip(axs, cmap_list):
         ax.imshow(gradient, aspect='auto', cmap=plt.get_cmap(name))
-        pos = list(ax.get_position().bounds)
-        x_text = pos[0] - 0.01
-        y_text = pos[1] + pos[3]/2.
-        fig.text(x_text, y_text, name, va='center', ha='right', fontsize=10)
+        ax.text(-0.01, 0.5, name, va='center', ha='right', fontsize=10,
+                transform=ax.transAxes)
 
     # Turn off *all* ticks & spines, not just the ones with colormaps.
     for ax in axs:
