Allow scalar height specification in plt.bar()

dstansby · dstansby · commit febc629ea74e · 2017-01-01T13:12:59.000Z
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -1876,8 +1876,8 @@ def bar(self, left, height, width=0.8, bottom=None, **kwargs):
         left : sequence of scalars
             the x coordinates of the left sides of the bars
 
-        height : sequence of scalars
-            the heights of the bars
+        height : scalar or sequence of scalars
+            the height(s) of the bars
 
         width : scalar or array-like, optional
             the width(s) of the bars
@@ -2025,6 +2025,8 @@ def make_iterable(x):
                 bottom = [0]
 
             nbars = len(left)
+            if len(height) == 1:
+                height *= nbars
             if len(width) == 1:
                 width *= nbars
             if len(bottom) == 1:
@@ -2047,6 +2049,8 @@ def make_iterable(x):
                 left *= nbars
             if len(height) == 1:
                 height *= nbars
+            if len(width) == 1:
+                width *= nbars
 
             tick_label_axis = self.yaxis
             tick_label_position = bottom
@@ -2077,7 +2081,7 @@ def make_iterable(x):
                              "be length %d or scalar" % nbars)
         if len(height) != nbars:
             raise ValueError("incompatible sizes: argument 'height' "
-                              "must be length %d or scalar" % nbars)
+                             "must be length %d or scalar" % nbars)
         if len(width) != nbars:
             raise ValueError("incompatible sizes: argument 'width' "
                              "must be length %d or scalar" % nbars)
