Changed orientation variable to string from boolean

ItsRLuo · QuLogic · commit a1ebd1f8adfd · 2020-08-12T23:43:43.000-04:00
Also modified documentation for the change.
Flipped thisx to thisy and vice versa when the orientation value is
vertical
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -2723,7 +2723,7 @@ def broken_barh(self, xranges, yrange, **kwargs):
 
     @_preprocess_data()
     def stem(self, *args, linefmt=None, markerfmt=None, basefmt=None, bottom=0,
-             label=None, use_line_collection=True, vertical=False):
+             label=None, use_line_collection=True, orientation='horizontal'):
         """
         Create a stem plot.
 
@@ -2774,8 +2774,9 @@ def stem(self, *args, linefmt=None, markerfmt=None, basefmt=None, bottom=0,
         basefmt : str, default: 'C3-' ('C2-' in classic mode)
             A format string defining the properties of the baseline.
 
-        vertical : bool, optional (False)
-            If 'True', will produce a vertically-oriented stem plot.
+        orientation : string, optional (horizontal)
+            If 'vertical', will produce a vertically-oriented stem plot,
+            else it will produce a horizontally-oriented stem plot.
 
         bottom : float, default: 0
             The y-position of the baseline.
@@ -2866,12 +2867,12 @@ def stem(self, *args, linefmt=None, markerfmt=None, basefmt=None, bottom=0,
             basestyle, basemarker, basecolor = _process_plot_format(basefmt)
 
         # Check if the user wants a vertical stem plot
-        if vertical:
+        if orientation == 'vertical':
             x, y = y, x
 
         # New behaviour in 3.1 is to use a LineCollection for the stemlines
         if use_line_collection:
-            if vertical:
+            if orientation == 'vertical':
                 stemlines = [((bottom, yi), (xi, yi)) for xi, yi in zip(x, y)]
             else:
                 stemlines = [((xi, bottom), (xi, yi)) for xi, yi in zip(x, y)]
@@ -2885,7 +2886,7 @@ def stem(self, *args, linefmt=None, markerfmt=None, basefmt=None, bottom=0,
         else:
             stemlines = []
             for xi, yi in zip(x, y):
-                if vertical:
+                if orientation == 'vertical':
                     xs = [bottom, xi]
                     ys = [yi, yi]
                 else:
@@ -2898,7 +2899,9 @@ def stem(self, *args, linefmt=None, markerfmt=None, basefmt=None, bottom=0,
 
         markerline, = self.plot(x, y, color=markercolor, linestyle=markerstyle,
                                 marker=markermarker, label="_nolegend_")
-        if vertical:
+
+        if orientation == 'vertical':
+            x, y = y, x
             baseline, = self.plot([bottom, bottom], [np.min(x), np.max(x)],
                                   color=basecolor, linestyle=basestyle,
                                   marker=basemarker, label="_nolegend_")
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -3018,12 +3018,13 @@ def stackplot(
 @_copy_docstring_and_deprecators(Axes.stem)
 def stem(
         *args, linefmt=None, markerfmt=None, basefmt=None, bottom=0,
-        label=None, use_line_collection=True, vertical=False,
-        data=None):
+        label=None, use_line_collection=True,
+        orientation='horizontal', data=None):
     return gca().stem(
         *args, linefmt=linefmt, markerfmt=markerfmt, basefmt=basefmt,
         bottom=bottom, label=label,
-        use_line_collection=use_line_collection, vertical=vertical,
+        use_line_collection=use_line_collection,
+        orientation=orientation,
         **({"data": data} if data is not None else {}))
 
 
