GH1441 Allow for axis argument in DataFrame.pct_change (#1450)

loicdiridollou · web-flow · commit 087504603cec · 2025-10-29T10:39:08.000+01:00
* GH1441 Allow for axis argument in DataFrame.pct_change

* GH1441 Allow for axis argument in DataFrame.pct_change
diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi
@@ -2227,6 +2227,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         fill_method: None = None,
         freq: Frequency | dt.timedelta | None = ...,
         fill_value: Scalar | NAType | None = ...,
+        axis: Axis | None = ...,
     ) -> Self: ...
     def pop(self, item: _str) -> Series: ...
     def pow(
diff --git a/tests/test_frame.py b/tests/test_frame.py
@@ -2828,6 +2828,10 @@ def test_dataframe_pct_change() -> None:
         pd.DataFrame,
     )
     check(assert_type(df.pct_change(fill_value=0), pd.DataFrame), pd.DataFrame)
+    check(assert_type(df.pct_change(axis=0), pd.DataFrame), pd.DataFrame)
+    check(assert_type(df.pct_change(axis=1), pd.DataFrame), pd.DataFrame)
+    check(assert_type(df.pct_change(axis="columns"), pd.DataFrame), pd.DataFrame)
+    check(assert_type(df.pct_change(axis="index"), pd.DataFrame), pd.DataFrame)
 
 
 def test_indexslice_setitem() -> None:

Original file line number	Diff line number	Diff line change
`@@ -2828,6 +2828,10 @@ def test_dataframe_pct_change() -> None:`
`2828`	`2828`	`pd.DataFrame,`
`2829`	`2829`	`)`
`2830`	`2830`	`check(assert_type(df.pct_change(fill_value=0), pd.DataFrame), pd.DataFrame)`
	`2831`	`+ check(assert_type(df.pct_change(axis=0), pd.DataFrame), pd.DataFrame)`
	`2832`	`+ check(assert_type(df.pct_change(axis=1), pd.DataFrame), pd.DataFrame)`
	`2833`	`+ check(assert_type(df.pct_change(axis="columns"), pd.DataFrame), pd.DataFrame)`
	`2834`	`+ check(assert_type(df.pct_change(axis="index"), pd.DataFrame), pd.DataFrame)`
`2831`	`2835`
`2832`	`2836`
`2833`	`2837`	`def test_indexslice_setitem() -> None:`