Skip to content

Commit c88d050

Browse files
authored
fix signature of frame.filter (#964)
items can be ListLike
1 parent 7e6aee4 commit c88d050

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

pandas-stubs/core/frame.pyi

+1-1
Original file line numberDiff line numberDiff line change
@@ -1711,7 +1711,7 @@ class DataFrame(NDFrame, OpsMixin):
17111711
) -> DataFrame: ...
17121712
def filter(
17131713
self,
1714-
items: list | None = ...,
1714+
items: ListLike | None = ...,
17151715
like: _str | None = ...,
17161716
regex: _str | None = ...,
17171717
axis: Axis | None = ...,

tests/test_frame.py

+2
Original file line numberDiff line numberDiff line change
@@ -319,6 +319,8 @@ def test_types_filter() -> None:
319319
df.filter(items=["col1"])
320320
df.filter(regex="co.*")
321321
df.filter(like="1")
322+
# [PR 964] Docs state `items` is `list-like`
323+
df.filter(items=("col2", "col2", 1, tuple([4])))
322324

323325

324326
def test_types_setting() -> None:

0 commit comments

Comments
 (0)