DOC: Use dict.items to iterate subplot_mosaic result

Author: QuLogic

galleries/users_explain/axes/arranging_axes.py

@@ -152,8 +152,8 @@ def annotate_axes(ax, text, fontsize=18):
 fig, axd = plt.subplot_mosaic([['upper left', 'upper right'],
                                ['lower left', 'lower right']],
                               figsize=(5.5, 3.5), layout="constrained")
-for k in axd:
-    annotate_axes(axd[k], f'axd["{k}"]', fontsize=14)
+for k, ax in axd.items():
+    annotate_axes(ax, f'axd[{k!r}]', fontsize=14)
 fig.suptitle('plt.subplot_mosaic()')
 
 # %%
@@ -200,8 +200,8 @@ def annotate_axes(ax, text, fontsize=18):
 fig, axd = plt.subplot_mosaic([['upper left', 'right'],
                                ['lower left', 'right']],
                               figsize=(5.5, 3.5), layout="constrained")
-for k in axd:
-    annotate_axes(axd[k], f'axd["{k}"]', fontsize=14)
+for k, ax in axd.items():
+    annotate_axes(ax, f'axd[{k!r}]', fontsize=14)
 fig.suptitle('plt.subplot_mosaic()')
 
 # %%
@@ -223,8 +223,8 @@ def annotate_axes(ax, text, fontsize=18):
                                ['lower left', 'right']],
                               gridspec_kw=gs_kw, figsize=(5.5, 3.5),
                               layout="constrained")
-for k in axd:
-    annotate_axes(axd[k], f'axd["{k}"]', fontsize=14)
+for k, ax in axd.items():
+    annotate_axes(ax, f'axd[{k!r}]', fontsize=14)
 fig.suptitle('plt.subplot_mosaic()')
 
 # %%
@@ -262,8 +262,8 @@ def annotate_axes(ax, text, fontsize=18):
           ['lower left', 'lower right']]
 
 fig, axd = plt.subplot_mosaic(outer, layout="constrained")
-for k in axd:
-    annotate_axes(axd[k], f'axd["{k}"]')
+for k, ax in axd.items():
+    annotate_axes(ax, f'axd[{k!r}]')
 
 # %%
 # Low-level and advanced grid methods

galleries/users_explain/axes/axes_scales.py

@@ -98,25 +98,23 @@
 # %%
 #
 
-todo = ['asinh', 'symlog', 'log', 'logit', ]
 fig, axs = plt.subplot_mosaic([['asinh', 'symlog'],
                                ['log', 'logit']], layout='constrained')
 
 x = np.arange(0, 1000)
 
-for td in todo:
-    ax = axs[td]
-    if td in ['asinh', 'symlog']:
+for name, ax in axs.items():
+    if name in ['asinh', 'symlog']:
         yy = x - np.mean(x)
-    elif td in ['logit']:
+    elif name in ['logit']:
         yy = (x-np.min(x))
         yy = yy / np.max(np.abs(yy))
     else:
         yy = x
 
     ax.plot(yy, yy)
-    ax.set_yscale(td)
-    ax.set_title(td)
+    ax.set_yscale(name)
+    ax.set_title(name)
 
 # %%
 # Optional arguments for scales
@@ -131,9 +129,8 @@
 fig, axs = plt.subplot_mosaic([['log', 'symlog']], layout='constrained',
                               figsize=(6.4, 3))
 
-for td in axs:
-    ax = axs[td]
-    if td in ['log']:
+for name, ax in axs.items():
+    if name in ['log']:
         ax.plot(x, x)
         ax.set_yscale('log', base=2)
         ax.set_title('log base=2')

galleries/users_explain/figure/figure_intro.rst

@@ -139,8 +139,8 @@ More complex grids can be achieved with `.pyplot.subplot_mosaic` (which wraps
 
     fig, axs = plt.subplot_mosaic([['A', 'right'], ['B', 'right']],
                                   figsize=(4, 3), layout='constrained')
-    for ax_name in axs:
-        axs[ax_name].text(0.5, 0.5, ax_name, ha='center', va='center')
+    for ax_name, ax in axs.items():
+        ax.text(0.5, 0.5, ax_name, ha='center', va='center')
 
 Sometimes we want to have a nested layout in a Figure, with two or more sets of
 Axes that do not share the same subplot grid.