BUG: Add min/max methods to ArrowExtensionArray GH#61311 (#61924)

Co-authored-by: Matthew Roeschke <[email protected]>

Author: skonda29

pandas/core/indexing.py

@@ -1626,9 +1626,17 @@ def _validate_key(self, key, axis: AxisInt) -> None:
             if not is_numeric_dtype(arr.dtype):
                 raise IndexError(f".iloc requires numeric indexers, got {arr}")
 
-            # check that the key does not exceed the maximum size of the index
-            if len(arr) and (arr.max() >= len_axis or arr.min() < -len_axis):
-                raise IndexError("positional indexers are out-of-bounds")
+            if len(arr):
+                if isinstance(arr.dtype, ExtensionDtype):
+                    arr_max = arr._reduce("max")
+                    arr_min = arr._reduce("min")
+                else:
+                    arr_max = np.max(arr)
+                    arr_min = np.min(arr)
+
+                # check that the key does not exceed the maximum size
+                if arr_max >= len_axis or arr_min < -len_axis:
+                    raise IndexError("positional indexers are out-of-bounds")
         else:
             raise ValueError(f"Can only index by ___location with a [{self._valid_types}]")
 

pandas/tests/indexing/test_iloc.py

@@ -1482,3 +1482,14 @@ def test_iloc_nullable_int64_size_1_nan(self):
         result = DataFrame({"a": ["test"], "b": [np.nan]})
         with pytest.raises(TypeError, match="Invalid value"):
             result.loc[:, "b"] = result.loc[:, "b"].astype("Int64")
+
+    def test_iloc_arrow_extension_array(self):
+        # GH#61311
+        pytest.importorskip("pyarrow")
+        df = DataFrame({"a": [1, 2], "c": [0, 2], "d": ["c", "a"]})
+        df_arrow = DataFrame(
+            {"a": [1, 2], "c": [0, 2], "d": ["c", "a"]}
+        ).convert_dtypes(dtype_backend="pyarrow")
+        expected = df.iloc[:, df["c"]]
+        result = df_arrow.iloc[:, df_arrow["c"]]
+        tm.assert_frame_equal(result, expected, check_dtype=False)