diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi index a081d5c09..482a2dcc3 100644 --- a/pandas-stubs/core/frame.pyi +++ b/pandas-stubs/core/frame.pyi @@ -89,7 +89,10 @@ class _iLocIndexerFrame(_iLocIndexer): @overload def __getitem__( self, - idx: IndexType | MaskType | tuple[IndexType | MaskType, IndexType | MaskType], + idx: IndexType + | MaskType + | tuple[IndexType | MaskType, IndexType | MaskType] + | tuple[slice], ) -> DataFrame: ... def __setitem__( self, diff --git a/tests/test_frame.py b/tests/test_frame.py index 27ced9263..9bd211d70 100644 --- a/tests/test_frame.py +++ b/tests/test_frame.py @@ -1152,6 +1152,14 @@ def test_iloc_npint() -> None: df.iloc[iloc] +# https://github.com/pandas-dev/pandas-stubs/issues/143 +def test_iloc_tuple() -> None: + df = pd.DataFrame({"Char": ["A", "B", "C"], "Number": [1, 2, 3]}) + df = df.iloc[ + 0:2, + ] + + def test_set_columns() -> None: # GH 73 df = pd.DataFrame({"a": [1, 2, 3], "b": [0.0, 1, 1]})