Skip to content

Commit bc754df

Browse files
Allow 'integer' in select_dtypes include argument (#968)
GH 927 Allow 'integer' in select_dtypes include argument
1 parent c88d050 commit bc754df

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

pandas-stubs/core/frame.pyi

+1
Original file line numberDiff line numberDiff line change
@@ -624,6 +624,7 @@ class DataFrame(NDFrame, OpsMixin):
624624
"number",
625625
"datetime64",
626626
"datetime",
627+
"integer",
627628
"timedelta",
628629
"timedelta64",
629630
"datetimetz",

tests/test_frame.py

+1
Original file line numberDiff line numberDiff line change
@@ -3226,6 +3226,7 @@ def test_convert_dtypes_dtype_backend() -> None:
32263226
def test_select_dtypes() -> None:
32273227
df = pd.DataFrame({"a": [1, 2] * 3, "b": [True, False] * 3, "c": [1.0, 2.0] * 3})
32283228
check(assert_type(df.select_dtypes("number"), pd.DataFrame), pd.DataFrame)
3229+
check(assert_type(df.select_dtypes("integer"), pd.DataFrame), pd.DataFrame)
32293230
check(assert_type(df.select_dtypes(np.number), pd.DataFrame), pd.DataFrame)
32303231
check(assert_type(df.select_dtypes(object), pd.DataFrame), pd.DataFrame)
32313232
check(assert_type(df.select_dtypes(include="bool"), pd.DataFrame), pd.DataFrame)

0 commit comments

Comments
 (0)