Skip to content

Commit 8186dfd

Browse files
committed
add limit_area parameter in ffill and bfill methods
1 parent 8d9be32 commit 8186dfd

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

pandas-stubs/core/frame.pyi

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1656,6 +1656,7 @@ class DataFrame(NDFrame, OpsMixin):
16561656
axis: Axis | None = ...,
16571657
inplace: Literal[True],
16581658
limit: int | None = ...,
1659+
limit_area: Literal["inside", "outside"] | None = ...,
16591660
downcast: dict | None = ...,
16601661
) -> None: ...
16611662
@overload
@@ -1665,6 +1666,7 @@ class DataFrame(NDFrame, OpsMixin):
16651666
axis: Axis | None = ...,
16661667
inplace: Literal[False] = ...,
16671668
limit: int | None = ...,
1669+
limit_area: Literal["inside", "outside"] | None = ...,
16681670
downcast: dict | None = ...,
16691671
) -> DataFrame: ...
16701672
def clip(
@@ -1736,6 +1738,7 @@ class DataFrame(NDFrame, OpsMixin):
17361738
axis: Axis | None = ...,
17371739
inplace: Literal[True],
17381740
limit: int | None = ...,
1741+
limit_area: Literal["inside", "outside"] | None = ...,
17391742
downcast: dict | None = ...,
17401743
) -> None: ...
17411744
@overload
@@ -1745,6 +1748,7 @@ class DataFrame(NDFrame, OpsMixin):
17451748
axis: Axis | None = ...,
17461749
inplace: Literal[False] = ...,
17471750
limit: int | None = ...,
1751+
limit_area: Literal["inside", "outside"] | None = ...,
17481752
downcast: dict | None = ...,
17491753
) -> DataFrame: ...
17501754
def filter(

tests/test_frame.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2328,15 +2328,25 @@ def test_types_ffill() -> None:
23282328
df = pd.DataFrame([[1, 2, 3]])
23292329
check(assert_type(df.ffill(), pd.DataFrame), pd.DataFrame)
23302330
check(assert_type(df.ffill(inplace=False), pd.DataFrame), pd.DataFrame)
2331+
check(
2332+
assert_type(df.ffill(inplace=False, limit_area="inside"), pd.DataFrame),
2333+
pd.DataFrame,
2334+
)
23312335
assert assert_type(df.ffill(inplace=True), None) is None
2336+
assert assert_type(df.ffill(inplace=True, limit_area="outside"), None) is None
23322337

23332338

23342339
def test_types_bfill() -> None:
23352340
# GH 44
23362341
df = pd.DataFrame([[1, 2, 3]])
23372342
check(assert_type(df.bfill(), pd.DataFrame), pd.DataFrame)
23382343
check(assert_type(df.bfill(inplace=False), pd.DataFrame), pd.DataFrame)
2344+
check(
2345+
assert_type(df.bfill(inplace=False, limit_area="inside"), pd.DataFrame),
2346+
pd.DataFrame,
2347+
)
23392348
assert assert_type(df.bfill(inplace=True), None) is None
2349+
assert assert_type(df.bfill(inplace=True, limit_area="outside"), None) is None
23402350

23412351

23422352
def test_types_replace() -> None:

0 commit comments

Comments
 (0)