BUG: Fix metadata propagation in squeeze and describe (#53538)

* BUG: Fix metadata propagation in squeeze and describe

* remove unnecessary param

Author: lithomas1

doc/source/whatsnew/v2.1.0.rst

@@ -484,6 +484,7 @@ Styler
 
 Metadata
 ^^^^^^^^
+- Fixed metadata propagation in :meth:`DataFrame.squeeze`, and :meth:`DataFrame.describe` (:issue:`28283`)
 - Fixed metadata propagation in :meth:`DataFrame.std` (:issue:`28283`)
 
 Other

pandas/core/generic.py

@@ -964,12 +964,15 @@ def squeeze(self, axis: Axis | None = None):
         1
         """
         axes = range(self._AXIS_LEN) if axis is None else (self._get_axis_number(axis),)
-        return self.iloc[
+        result = self.iloc[
             tuple(
                 0 if i in axes and len(a) == 1 else slice(None)
                 for i, a in enumerate(self.axes)
             )
         ]
+        if isinstance(result, NDFrame):
+            result = result.__finalize__(self, method="squeeze")
+        return result
 
     # ----------------------------------------------------------------------
     # Rename
@@ -11144,7 +11147,7 @@ def describe(
             include=include,
             exclude=exclude,
             percentiles=percentiles,
-        )
+        ).__finalize__(self, method="describe")
 
     @final
     def pct_change(

pandas/tests/generic/test_finalize.py

@@ -223,10 +223,8 @@
     ),
     (pd.DataFrame, frame_mi_data, operator.methodcaller("droplevel", "A")),
     (pd.DataFrame, frame_data, operator.methodcaller("pop", "A")),
-    pytest.param(
-        (pd.DataFrame, frame_data, operator.methodcaller("squeeze")),
-        marks=not_implemented_mark,
-    ),
+    # Squeeze on columns, otherwise we'll end up with a scalar
+    (pd.DataFrame, frame_data, operator.methodcaller("squeeze", axis="columns")),
     (pd.Series, ([1, 2],), operator.methodcaller("squeeze")),
     (pd.Series, ([1, 2],), operator.methodcaller("rename_axis", index="a")),
     (pd.DataFrame, frame_data, operator.methodcaller("rename_axis", columns="a")),
@@ -350,14 +348,8 @@
         ({"A": [1, 1, 1, 1]}, pd.date_range("2000", periods=4)),
         operator.methodcaller("tz_localize", "CET"),
     ),
-    pytest.param(
-        (pd.Series, ([1, 2],), operator.methodcaller("describe")),
-        marks=not_implemented_mark,
-    ),
-    pytest.param(
-        (pd.DataFrame, frame_data, operator.methodcaller("describe")),
-        marks=not_implemented_mark,
-    ),
+    (pd.Series, ([1, 2],), operator.methodcaller("describe")),
+    (pd.DataFrame, frame_data, operator.methodcaller("describe")),
     (pd.Series, ([1, 2],), operator.methodcaller("pct_change")),
     (pd.DataFrame, frame_data, operator.methodcaller("pct_change")),
     (pd.Series, ([1],), operator.methodcaller("transform", lambda x: x - x.min())),
@@ -737,7 +729,6 @@ def test_groupby_finalize(obj, method):
         lambda x: x.agg("sem"),
         lambda x: x.agg("size"),
         lambda x: x.agg("ohlc"),
-        lambda x: x.agg("describe"),
     ],
 )
 @not_implemented_mark