Backport PR pandas-dev#48472: PERF: keep using ObjectEngine for Exten…

…sionArrays for 1.5
meeseeksmachine · Sep 9, 2022 · 716beae · 716beae
1 parent 54c1aa2
commit 716beae
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pandas/core/indexes/base.py b/pandas/core/indexes/base.py
@@ -5154,6 +5154,9 @@ def _get_engine_target(self) -> ArrayLike:
         if isinstance(vals, StringArray):
             # GH#45652 much more performant than ExtensionEngine
             return vals._ndarray
+        if type(self) is Index and isinstance(self._values, ExtensionArray):
+            # TODO(ExtensionIndex): remove special-case, just use self._values
+            return self._values.astype(object)
         return vals
 
     def _from_join_target(self, result: np.ndarray) -> ArrayLike: