ENH: Enabled prefix, suffix, and sep to DataFrame. shift with iterable periods (#61696)

Author: RUTUPARNk

pandas/core/frame.py

@@ -5866,7 +5866,9 @@ def shift(
         freq: Frequency | None = None,
         axis: Axis = 0,
         fill_value: Hashable = lib.no_default,
+        prefix: str | None = None,
         suffix: str | None = None,
+        sep: str ="_",
     ) -> DataFrame:
         if freq is not None and fill_value is not lib.no_default:
             # GH#53832
@@ -5899,11 +5901,17 @@ def shift(
                 shifted_dataframes.append(
                     super()
                     .shift(periods=period, freq=freq, axis=axis, fill_value=fill_value)
+                    .add_prefix(f"{prefix}{sep}" if prefix else "")
                     .add_suffix(f"{suffix}_{period}" if suffix else f"_{period}")
+                    .rename(
+                        columns=lambda col: col.replace(f"{sep}{sep}", sep)
+                    )
                 )
             return concat(shifted_dataframes, axis=1)
         elif suffix:
             raise ValueError("Cannot specify `suffix` if `periods` is an int.")
+        elif prefix:
+            raise ValueError("Cannot specify `prefix` if `periods` is an int.")
         periods = cast(int, periods)
 
         ncols = len(self.columns)

pandas/tests/frame/methods/test_shift.py

@@ -725,6 +725,16 @@ def test_shift_with_iterable_check_other_arguments(self):
         expected = DataFrame({"a_suffix_0": [1, 2], "a_suffix_1": [np.nan, 1.0]})
         tm.assert_frame_equal(shifted, expected)
 
+        # test prefix and sep
+        shifted = df[["a"]].shift(shifts, prefix="pre", suffix="suf", sep="-")
+        expected = DataFrame(
+            {
+                "pre-a-suf-0": [1, 2],
+                "pre-a-suf-1": [np.nan, 1.0]
+            }
+        )
+        tm.assert_frame_equal(shifted, expected)
+
         # check bad inputs when doing multiple shifts
         msg = "If `periods` contains multiple shifts, `axis` cannot be 1."
         with pytest.raises(ValueError, match=msg):