Add unit support to stem

dstansby · dstansby · commit b6760657ae8d · 2018-06-28T10:29:18.000+01:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -2364,6 +2364,9 @@ def stem(self, *args, linefmt=None, markerfmt=None, basefmt=None,
             x = y
             y = np.asarray(args[0], dtype=float)
             args = args[1:]
+        self._process_unit_info(xdata=x, ydata=y, kwargs=kwargs)
+        x = self.convert_xunits(x)
+        y = self.convert_yunits(y)
 
         # defaults for formats
         if linefmt is None:
