optimize BlockManager.column_arrays

pandas-dev · Apr 21, 2021 · 3c9aa99 · 3c9aa99
1 parent 3b7e2a1
commit 3c9aa99
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/pandas/core/internals/managers.py b/pandas/core/internals/managers.py
@@ -1139,7 +1139,16 @@ def iget_values(self, i: int) -> ArrayLike:
 
     @property
     def column_arrays(self) -> list[ArrayLike]:
-        return [self.iget_values(i) for i in range(len(self.items))]
+        arrays = [np.asarray(arr) for arr in self.arrays]
+        result = []
+        for i in range(len(self.items)):
+            arr = arrays[self.blknos[i]]
+            if arr.ndim == 2:
+                values = arr[self.blklocs[i]]
+            else:
+                values = arr
+            result.append(values)
+        return result
 
     def iset(self, loc: int | slice | np.ndarray, value: ArrayLike):
         """