Move colorbar label to constructor call in examples.

anntzer · anntzer · commit 000c193f8e68 · 2019-12-30T22:06:19.000+01:00
diff --git a/examples/specialty_plots/leftventricle_bulleye.py b/examples/specialty_plots/leftventricle_bulleye.py
@@ -149,17 +149,15 @@ def bullseye_plot(ax, data, seg_bold=None, cmap=None, norm=None):
 norm = mpl.colors.Normalize(vmin=1, vmax=17)
 # Create an empty ScalarMappable to set the colorbar's colormap and norm.
 # The following gives a basic continuous colorbar with ticks and labels.
-cb1 = fig.colorbar(mpl.cm.ScalarMappable(cmap=cmap, norm=norm),
-                   cax=axl, orientation='horizontal')
-cb1.set_label('Some Units')
+fig.colorbar(mpl.cm.ScalarMappable(cmap=cmap, norm=norm),
+             cax=axl, orientation='horizontal', label='Some Units')
 
 
 # And again for the second colorbar.
 cmap2 = mpl.cm.cool
 norm2 = mpl.colors.Normalize(vmin=1, vmax=17)
-cb2 = fig.colorbar(mpl.cm.ScalarMappable(cmap=cmap2, norm=norm2),
-                   cax=axl2, orientation='horizontal')
-cb2.set_label('Some other units')
+fig.colorbar(mpl.cm.ScalarMappable(cmap=cmap2, norm=norm2),
+             cax=axl2, orientation='horizontal', label='Some other units')
 
 
 # The second example illustrates the use of a ListedColormap, a
@@ -173,15 +171,15 @@ def bullseye_plot(ax, data, seg_bold=None, cmap=None, norm=None):
 # monotonically increasing.
 bounds = [2, 3, 7, 9, 15]
 norm3 = mpl.colors.BoundaryNorm(bounds, cmap3.N)
-cb3 = fig.colorbar(mpl.cm.ScalarMappable(cmap=cmap3, norm=norm3),
-                   cax=axl3,
-                   # to use 'extend', you must specify two extra boundaries:
-                   boundaries=[0] + bounds + [18],
-                   extend='both',
-                   ticks=bounds,  # optional
-                   spacing='proportional',
-                   orientation='horizontal')
-cb3.set_label('Discrete intervals, some other units')
+fig.colorbar(mpl.cm.ScalarMappable(cmap=cmap3, norm=norm3),
+             cax=axl3,
+             # to use 'extend', you must specify two extra boundaries:
+             boundaries=[0] + bounds + [18],
+             extend='both',
+             ticks=bounds,  # optional
+             spacing='proportional',
+             orientation='horizontal',
+             label='Discrete intervals, some other units')
 
 
 # Create the 17 segment model
diff --git a/tutorials/colors/colorbar_only.py b/tutorials/colors/colorbar_only.py
@@ -35,9 +35,8 @@
 cmap = mpl.cm.cool
 norm = mpl.colors.Normalize(vmin=5, vmax=10)
 
-cb1 = fig.colorbar(mpl.cm.ScalarMappable(norm=norm, cmap=cmap),
-                   cax=ax, orientation='horizontal')
-cb1.set_label('Some Units')
+fig.colorbar(mpl.cm.ScalarMappable(norm=norm, cmap=cmap),
+             cax=ax, orientation='horizontal', label='Some Units')
 
 ###############################################################################
 # Discrete intervals colorbar
@@ -70,15 +69,16 @@
 
 bounds = [1, 2, 4, 7, 8]
 norm = mpl.colors.BoundaryNorm(bounds, cmap.N)
-cb2 = fig.colorbar(
+fig.colorbar(
     mpl.cm.ScalarMappable(cmap=cmap, norm=norm),
     cax=ax,
     boundaries=[0] + bounds + [13],
     extend='both',
     ticks=bounds,
     spacing='proportional',
-    orientation='horizontal')
-cb2.set_label('Discrete intervals, some other units')
+    orientation='horizontal',
+    label='Discrete intervals, some other units',
+)
 
 ###############################################################################
 # Colorbar with custom extension lengths
@@ -98,15 +98,16 @@
 
 bounds = [-1.0, -0.5, 0.0, 0.5, 1.0]
 norm = mpl.colors.BoundaryNorm(bounds, cmap.N)
-cb3 = fig.colorbar(
+fig.colorbar(
     mpl.cm.ScalarMappable(cmap=cmap, norm=norm),
     cax=ax,
     boundaries=[-10] + bounds + [10],
     extend='both',
     extendfrac='auto',
     ticks=bounds,
     spacing='uniform',
-    orientation='horizontal')
-cb3.set_label('Custom extension lengths, some other units')
+    orientation='horizontal',
+    label='Custom extension lengths, some other units',
+)
 
 plt.show()
