PERF: avoid validating indices repeatedly in ArrayManager.reindex_ind…

…exer (#40248)
pandas-dev · Mar 5, 2021 · bf270f6 · bf270f6
1 parent ee50197
commit bf270f6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pandas/core/internals/array_manager.py b/pandas/core/internals/array_manager.py
@@ -73,7 +73,10 @@
     extract_array,
     sanitize_array,
 )
-from pandas.core.indexers import maybe_convert_indices
+from pandas.core.indexers import (
+    maybe_convert_indices,
+    validate_indices,
+)
 from pandas.core.indexes.api import (
     Index,
     ensure_index,
@@ -965,8 +968,9 @@ def _reindex_indexer(
                 new_arrays.append(arr)
 
         else:
+            validate_indices(indexer, len(self._axes[0]))
             new_arrays = [
-                algos.take(
+                take_nd(
                     arr,
                     indexer,
                     allow_fill=True,