Merge pull request #1819 from ksunden/mpl_units_array

Short circuit for pint arrays
hgrecco · Jul 17, 2023 · 4cca8ab · 4cca8ab
2 parents 2d649ea + 2be029f
commit 4cca8ab
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/CHANGES b/CHANGES
@@ -8,6 +8,8 @@ Pint Changelog
   (PR #1805)
 - Documented to_preferred and created added an autoautoconvert_to_preferred registry option.
   (PR #1803)
+- Optimize matplotlib unit conversion for Quantity arrays
+  (PR #1819)
 
 
 0.22 (2023-05-25)

diff --git a/pint/matplotlib.py b/pint/matplotlib.py
@@ -34,6 +34,9 @@ def __init__(self, registry):
 
     def convert(self, value, unit, axis):
         """Convert :`Quantity` instances for matplotlib to use."""
+        # Short circuit for arrays
+        if hasattr(value, "units"):
+            return value.to(unit).magnitude
         if iterable(value):
             return [self._convert_value(v, unit, axis) for v in value]