REF: share common code in ArrayManager.get_numeric_data/get_bool_data (#40030)

Author: jorisvandenbossche

pandas/core/internals/array_manager.py

@@ -480,31 +480,34 @@ def is_view(self) -> bool:
     def is_single_block(self) -> bool:
         return False
 
+    def _get_data_subset(self, predicate: Callable) -> ArrayManager:
+        indices = [i for i, arr in enumerate(self.arrays) if predicate(arr)]
+        arrays = [self.arrays[i] for i in indices]
+        # TODO copy?
+        new_axes = [self._axes[0], self._axes[1][np.array(indices, dtype="int64")]]
+        return type(self)(arrays, new_axes, verify_integrity=False)
+
     def get_bool_data(self, copy: bool = False) -> ArrayManager:
         """
+        Select columns that are bool-dtype.
+
         Parameters
         ----------
         copy : bool, default False
             Whether to copy the blocks
         """
-        mask = np.array([is_bool_dtype(t) for t in self.get_dtypes()], dtype="object")
-        arrays = [self.arrays[i] for i in np.nonzero(mask)[0]]
-        # TODO copy?
-        new_axes = [self._axes[0], self._axes[1][mask]]
-        return type(self)(arrays, new_axes)
+        return self._get_data_subset(lambda arr: is_bool_dtype(arr.dtype))
 
     def get_numeric_data(self, copy: bool = False) -> ArrayManager:
         """
+        Select columns that have a numeric dtype.
+
         Parameters
         ----------
         copy : bool, default False
             Whether to copy the blocks
         """
-        mask = np.array([is_numeric_dtype(t) for t in self.get_dtypes()])
-        arrays = [self.arrays[i] for i in np.nonzero(mask)[0]]
-        # TODO copy?
-        new_axes = [self._axes[0], self._axes[1][mask]]
-        return type(self)(arrays, new_axes)
+        return self._get_data_subset(lambda arr: is_numeric_dtype(arr.dtype))
 
     def copy(self: T, deep=True) -> T:
         """