Add unit support to stem

dstansby · dstansby · commit 3de867b00eb1 · 2017-10-26T21:38:47.000+01:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -2447,6 +2447,9 @@ def stem(self, *args, **kwargs):
             # The second array doesn't make sense, or it doesn't exist
             second = np.arange(len(y))
             x = second
+        self._process_unit_info(xdata=x, ydata=y, kwargs=kwargs)
+        x = self.convert_xunits(x)
+        y = self.convert_yunits(y)
 
         # Popping some defaults
         try:
