diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi index ef52cdaf9..74f48144d 100644 --- a/pandas-stubs/core/frame.pyi +++ b/pandas-stubs/core/frame.pyi @@ -202,7 +202,7 @@ class _LocIndexerFrame(_LocIndexer): @overload def __setitem__( self, - idx: MaskType | StrLike | _IndexSliceTuple | list[ScalarT], + idx: MaskType | StrLike | _IndexSliceTuple | list[ScalarT] | IndexingInt, value: Scalar | NAType | NaTType | ArrayLike | Series | DataFrame | list | None, ) -> None: ... @overload diff --git a/tests/test_frame.py b/tests/test_frame.py index be7e9fd9f..e8dc8042b 100644 --- a/tests/test_frame.py +++ b/tests/test_frame.py @@ -2290,6 +2290,16 @@ def test_loc_set() -> None: df.loc["a"] = [3, 4] +def test_loc_int_set() -> None: + df = pd.DataFrame({1: [1, 2], 2: [3, 4]}) + df.loc[1] = [3, 4] + df.loc[np.int_(1)] = pd.Series([1, 2]) + df.loc[np.uint(1)] = pd.Series([1, 2]) + df.loc[np.int8(1)] = pd.Series([1, 2]) + df.loc[np.int32(1)] = [2, 3] + df.loc[np.uint64(1)] = [2, 3] + + def test_loclist() -> None: # GH 189 df = pd.DataFrame({1: [1, 2], None: 5}, columns=pd.Index([1, None], dtype=object))