Skip to content

Commit 94df875

Browse files
authored
Fix: dataframe.loc int label could be warn (#816)
1 parent 2710616 commit 94df875

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

pandas-stubs/core/frame.pyi

+1-1
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ class _LocIndexerFrame(_LocIndexer):
202202
@overload
203203
def __setitem__(
204204
self,
205-
idx: MaskType | StrLike | _IndexSliceTuple | list[ScalarT],
205+
idx: MaskType | StrLike | _IndexSliceTuple | list[ScalarT] | IndexingInt,
206206
value: Scalar | NAType | NaTType | ArrayLike | Series | DataFrame | list | None,
207207
) -> None: ...
208208
@overload

tests/test_frame.py

+10
Original file line numberDiff line numberDiff line change
@@ -2290,6 +2290,16 @@ def test_loc_set() -> None:
22902290
df.loc["a"] = [3, 4]
22912291

22922292

2293+
def test_loc_int_set() -> None:
2294+
df = pd.DataFrame({1: [1, 2], 2: [3, 4]})
2295+
df.loc[1] = [3, 4]
2296+
df.loc[np.int_(1)] = pd.Series([1, 2])
2297+
df.loc[np.uint(1)] = pd.Series([1, 2])
2298+
df.loc[np.int8(1)] = pd.Series([1, 2])
2299+
df.loc[np.int32(1)] = [2, 3]
2300+
df.loc[np.uint64(1)] = [2, 3]
2301+
2302+
22932303
def test_loclist() -> None:
22942304
# GH 189
22952305
df = pd.DataFrame({1: [1, 2], None: 5}, columns=pd.Index([1, None], dtype=object))

0 commit comments

Comments
 (0)