ENH: Add numeric_only to window ops

Author: rhshadrach

pandas/core/window/doc.py

@@ -29,6 +29,14 @@ def create_section_header(header: str) -> str:
     """
 ).replace("\n", "", 1)
 
+kwargs_numeric_only = dedent(
+    """
+    numeric_only : bool, default False
+        Include only float, int, boolean columns. If None, will attempt to use
+        everything, then use only numeric data.
+    """
+)
+
 args_compat = dedent(
     """
     *args

pandas/core/window/ewm.py

@@ -45,6 +45,7 @@
     args_compat,
     create_section_header,
     kwargs_compat,
+    kwargs_numeric_only,
     numba_notes,
     template_header,
     template_returns,
@@ -518,6 +519,7 @@ def aggregate(self, func, *args, **kwargs):
     @doc(
         template_header,
         create_section_header("Parameters"),
+        kwargs_numeric_only,
         args_compat,
         window_agg_numba_parameters(),
         kwargs_compat,
@@ -531,7 +533,14 @@ def aggregate(self, func, *args, **kwargs):
         aggregation_description="(exponential weighted moment) mean",
         agg_method="mean",
     )
-    def mean(self, *args, engine=None, engine_kwargs=None, **kwargs):
+    def mean(
+        self,
+        numeric_only: bool = False,
+        *args,
+        engine=None,
+        engine_kwargs=None,
+        **kwargs,
+    ):
         if maybe_use_numba(engine):
             if self.method == "single":
                 func = generate_numba_ewm_func
@@ -560,13 +569,14 @@ def mean(self, *args, engine=None, engine_kwargs=None, **kwargs):
                 deltas=deltas,
                 normalize=True,
             )
-            return self._apply(window_func)
+            return self._apply(window_func, numeric_only=numeric_only)
         else:
             raise ValueError("engine must be either 'numba' or 'cython'")
 
     @doc(
         template_header,
         create_section_header("Parameters"),
+        kwargs_numeric_only,
         args_compat,
         window_agg_numba_parameters(),
         kwargs_compat,
@@ -580,7 +590,14 @@ def mean(self, *args, engine=None, engine_kwargs=None, **kwargs):
         aggregation_description="(exponential weighted moment) sum",
         agg_method="sum",
     )
-    def sum(self, *args, engine=None, engine_kwargs=None, **kwargs):
+    def sum(
+        self,
+        numeric_only: bool = False,
+        *args,
+        engine=None,
+        engine_kwargs=None,
+        **kwargs,
+    ):
         if not self.adjust:
             raise NotImplementedError("sum is not implemented with adjust=False")
         if maybe_use_numba(engine):
@@ -611,7 +628,7 @@ def sum(self, *args, engine=None, engine_kwargs=None, **kwargs):
                 deltas=deltas,
                 normalize=False,
             )
-            return self._apply(window_func)
+            return self._apply(window_func, numeric_only=numeric_only)
         else:
             raise ValueError("engine must be either 'numba' or 'cython'")
 
@@ -624,6 +641,7 @@ def sum(self, *args, engine=None, engine_kwargs=None, **kwargs):
             Use a standard estimation bias correction.
         """
         ).replace("\n", "", 1),
+        kwargs_numeric_only,
         args_compat,
         kwargs_compat,
         create_section_header("Returns"),
@@ -634,9 +652,9 @@ def sum(self, *args, engine=None, engine_kwargs=None, **kwargs):
         aggregation_description="(exponential weighted moment) standard deviation",
         agg_method="std",
     )
-    def std(self, bias: bool = False, *args, **kwargs):
+    def std(self, bias: bool = False, numeric_only: bool = False, *args, **kwargs):
         nv.validate_window_func("std", args, kwargs)
-        return zsqrt(self.var(bias=bias, **kwargs))
+        return zsqrt(self.var(bias=bias, numeric_only=numeric_only, **kwargs))
 
     def vol(self, bias: bool = False, *args, **kwargs):
         warnings.warn(
@@ -658,6 +676,7 @@ def vol(self, bias: bool = False, *args, **kwargs):
             Use a standard estimation bias correction.
         """
         ).replace("\n", "", 1),
+        kwargs_numeric_only,
         args_compat,
         kwargs_compat,
         create_section_header("Returns"),
@@ -668,7 +687,7 @@ def vol(self, bias: bool = False, *args, **kwargs):
         aggregation_description="(exponential weighted moment) variance",
         agg_method="var",
     )
-    def var(self, bias: bool = False, *args, **kwargs):
+    def var(self, bias: bool = False, numeric_only: bool = False, *args, **kwargs):
         nv.validate_window_func("var", args, kwargs)
         window_func = window_aggregations.ewmcov
         wfunc = partial(
@@ -682,7 +701,7 @@ def var(self, bias: bool = False, *args, **kwargs):
         def var_func(values, begin, end, min_periods):
             return wfunc(values, begin, end, min_periods, values)
 
-        return self._apply(var_func)
+        return self._apply(var_func, numeric_only=numeric_only)
 
     @doc(
         template_header,
@@ -703,6 +722,7 @@ def var_func(values, begin, end, min_periods):
             Use a standard estimation bias correction.
         """
         ).replace("\n", "", 1),
+        kwargs_numeric_only,
         kwargs_compat,
         create_section_header("Returns"),
         template_returns,
@@ -717,6 +737,7 @@ def cov(
         other: DataFrame | Series | None = None,
         pairwise: bool | None = None,
         bias: bool = False,
+        numeric_only: bool = False,
         **kwargs,
     ):
         from pandas import Series
@@ -752,7 +773,9 @@ def cov_func(x, y):
             )
             return Series(result, index=x.index, name=x.name)
 
-        return self._apply_pairwise(self._selected_obj, other, pairwise, cov_func)
+        return self._apply_pairwise(
+            self._selected_obj, other, pairwise, cov_func, numeric_only
+        )
 
     @doc(
         template_header,
@@ -771,6 +794,7 @@ def cov_func(x, y):
             observations will be used.
         """
         ).replace("\n", "", 1),
+        kwargs_numeric_only,
         kwargs_compat,
         create_section_header("Returns"),
         template_returns,
@@ -784,6 +808,7 @@ def corr(
         self,
         other: DataFrame | Series | None = None,
         pairwise: bool | None = None,
+        numeric_only: bool = False,
         **kwargs,
     ):
         from pandas import Series
@@ -825,7 +850,9 @@ def _cov(X, Y):
                 result = cov / zsqrt(x_var * y_var)
             return Series(result, index=x.index, name=x.name)
 
-        return self._apply_pairwise(self._selected_obj, other, pairwise, cov_func)
+        return self._apply_pairwise(
+            self._selected_obj, other, pairwise, cov_func, numeric_only
+        )
 
 
 class ExponentialMovingWindowGroupby(BaseWindowGroupby, ExponentialMovingWindow):
@@ -921,6 +948,7 @@ def corr(
         self,
         other: DataFrame | Series | None = None,
         pairwise: bool | None = None,
+        numeric_only: bool = False,
         **kwargs,
     ):
         return NotImplementedError
@@ -930,6 +958,7 @@ def cov(
         other: DataFrame | Series | None = None,
         pairwise: bool | None = None,
         bias: bool = False,
+        numeric_only: bool = False,
         **kwargs,
     ):
         return NotImplementedError