BUG: Allow numeric ExtensionDtypes in DataFrame.select_dtypes

doc/source/whatsnew/v1.2.0.rst

@@ -623,6 +623,7 @@ Numeric
 - Bug in :meth:`DataFrame.std` with ``timedelta64`` dtype and ``skipna=False`` (:issue:`37392`)
 - Bug in :meth:`DataFrame.min` and :meth:`DataFrame.max` with ``datetime64`` dtype and ``skipna=False`` (:issue:`36907`)
 - Bug in :meth:`DataFrame.idxmax` and :meth:`DataFrame.idxmin` with mixed dtypes incorrectly raising ``TypeError`` (:issue:`38195`)
+- Bug in :meth:`DataFrame.select_dtypes` with ``include=np.number`` now retains numeric ``ExtensionDtype`` columns (:issue:`35340`)
 
 Conversion
 ^^^^^^^^^^

pandas/core/frame.py

@@ -3711,12 +3711,15 @@ def extract_unique_dtypes_from_dtypes_set(
             extracted_dtypes = [
                 unique_dtype
                 for unique_dtype in unique_dtypes
-                # error: Argument 1 to "tuple" has incompatible type
-                # "FrozenSet[Union[ExtensionDtype, str, Any, Type[str],
-                # Type[float], Type[int], Type[complex], Type[bool]]]";
-                # expected "Iterable[Union[type, Tuple[Any, ...]]]"
-                if issubclass(
-                    unique_dtype.type, tuple(dtypes_set)  # type: ignore[arg-type]
+                if (
+                    issubclass(
+                        unique_dtype.type, tuple(dtypes_set)  # type: ignore[arg-type]
+                    )
+                    or (
+                        np.number in dtypes_set
+                        and hasattr(unique_dtype, "_is_numeric")  # is an extensionarray
+                        and unique_dtype._is_numeric
+                    )
                 )
             ]
             return extracted_dtypes

pandas/tests/frame/methods/test_select_dtypes.py

@@ -1,9 +1,46 @@
 import numpy as np
 import pytest
 
+from pandas.core.dtypes.dtypes import ExtensionDtype
+
 import pandas as pd
 from pandas import DataFrame, Timestamp
 import pandas._testing as tm
+from pandas.core.arrays import ExtensionArray
+
+
+class DummyDtype(ExtensionDtype):
+    type = int
+
+    def __init__(self, numeric):
+        self._numeric = numeric
+
+    @property
+    def name(self):
+        return "Dummy"
+
+    @property
+    def _is_numeric(self):
+        return self._numeric
+
+
+class DummyArray(ExtensionArray):
+    def __init__(self, data, dtype):
+        self.data = data
+        self._dtype = dtype
+
+    def __array__(self, dtype):
+        return self.data
+
+    @property
+    def dtype(self):
+        return self._dtype
+
+    def __len__(self) -> int:
+        return len(self.data)
+
+    def __getitem__(self, item):
+        pass
 
 
 class TestSelectDtypes:
@@ -322,3 +359,14 @@ def test_select_dtypes_typecodes(self):
         expected = df
         FLOAT_TYPES = list(np.typecodes["AllFloat"])
         tm.assert_frame_equal(df.select_dtypes(FLOAT_TYPES), expected)
+
+    def test_select_dtypes_numeric(self):
+        # GH 35340
+
+        da = DummyArray([1, 2], dtype=DummyDtype(numeric=True))
+        df = DataFrame(da)
+        assert df.select_dtypes(np.number).shape == df.shape
+
+        da = DummyArray([1, 2], dtype=DummyDtype(numeric=False))
+        df = DataFrame(da)
+        assert df.select_dtypes(np.number).shape != df.shape