update algo.take to solve pandas-dev#59177

Author: Laurent Mutricy

pandas/core/algorithms.py

@@ -1164,8 +1164,8 @@ def take(
     if not isinstance(arr, (np.ndarray, ABCExtensionArray, ABCIndex, ABCSeries)):
         # GH#52981
         raise TypeError(
-            "pd.api.extensions.take requires a numpy.ndarray, "
-            f"ExtensionArray, Index, or Series, got {type(arr).__name__}."
+            "pd.api.extensions.take requires a numpy.ndarray, ExtensionArray, "
+            f"Index, Series, or NumpyExtensionArray got {type(arr).__name__}."
         )
 
     indices = ensure_platform_int(indices)

pandas/core/dtypes/generic.py

@@ -139,7 +139,14 @@ def _subclasscheck(cls, inst) -> bool:
         "ABCExtensionArray",
         "_typ",
         # Note: IntervalArray and SparseArray are included bc they have _typ="extension"
-        {"extension", "categorical", "periodarray", "datetimearray", "timedeltaarray"},
+        {
+            "extension",
+            "categorical",
+            "periodarray",
+            "datetimearray",
+            "timedeltaarray",
+            "npy_extension",
+        },
     ),
 )
 ABCNumpyExtensionArray = cast(

pandas/tests/test_take.py

@@ -5,6 +5,7 @@
 
 from pandas._libs import iNaT
 
+from pandas import array
 import pandas._testing as tm
 import pandas.core.algorithms as algos
 
@@ -307,3 +308,10 @@ def test_take_coerces_list(self):
         )
         with pytest.raises(TypeError, match=msg):
             algos.take(arr, [0, 0])
+
+    def test_take_NumpyExtensionArray(self):
+        # GH#59177
+        arr = array([1 + 1j, 2, 3])  # NumpyEADtype('complex128') (NumpyExtensionArray)
+        assert algos.take(arr, [2]) == 2
+        arr = array([1, 2, 3])  # Int64Dtype() (ExtensionArray)
+        assert algos.take(arr, [2]) == 2