diff --git a/pandas/core/ops/docstrings.py b/pandas/core/ops/docstrings.py index e3db65f11a332..203ea3946d1b2 100644 --- a/pandas/core/ops/docstrings.py +++ b/pandas/core/ops/docstrings.py @@ -34,6 +34,7 @@ def _make_flex_doc(op_name, typ): op_name=op_name, equiv=equiv, reverse=op_desc["reverse"], + series_returns=op_desc["series_returns"], ) if op_desc["series_examples"]: doc = doc_no_examples + op_desc["series_examples"] @@ -233,6 +234,10 @@ def _make_flex_doc(op_name, typ): dtype: float64 """ +_returns_series = """Series\n The result of the operation.""" + +_returns_tuple = """2-Tuple of Series\n The result of the operation.""" + _op_descriptions: Dict[str, Dict[str, Optional[str]]] = { # Arithmetic Operators "add": { @@ -240,18 +245,21 @@ def _make_flex_doc(op_name, typ): "desc": "Addition", "reverse": "radd", "series_examples": _add_example_SERIES, + "series_returns": _returns_series, }, "sub": { "op": "-", "desc": "Subtraction", "reverse": "rsub", "series_examples": _sub_example_SERIES, + "series_returns": _returns_series, }, "mul": { "op": "*", "desc": "Multiplication", "reverse": "rmul", "series_examples": _mul_example_SERIES, + "series_returns": _returns_series, "df_examples": None, }, "mod": { @@ -259,12 +267,14 @@ def _make_flex_doc(op_name, typ): "desc": "Modulo", "reverse": "rmod", "series_examples": _mod_example_SERIES, + "series_returns": _returns_series, }, "pow": { "op": "**", "desc": "Exponential power", "reverse": "rpow", "series_examples": _pow_example_SERIES, + "series_returns": _returns_series, "df_examples": None, }, "truediv": { @@ -272,6 +282,7 @@ def _make_flex_doc(op_name, typ): "desc": "Floating division", "reverse": "rtruediv", "series_examples": _div_example_SERIES, + "series_returns": _returns_series, "df_examples": None, }, "floordiv": { @@ -279,6 +290,7 @@ def _make_flex_doc(op_name, typ): "desc": "Integer division", "reverse": "rfloordiv", "series_examples": _floordiv_example_SERIES, + "series_returns": _returns_series, "df_examples": None, }, "divmod": { @@ -286,29 +298,51 @@ def _make_flex_doc(op_name, typ): "desc": "Integer division and modulo", "reverse": "rdivmod", "series_examples": None, + "series_returns": _returns_tuple, "df_examples": None, }, # Comparison Operators - "eq": {"op": "==", "desc": "Equal to", "reverse": None, "series_examples": None}, + "eq": { + "op": "==", + "desc": "Equal to", + "reverse": None, + "series_examples": None, + "series_returns": _returns_series, + }, "ne": { "op": "!=", "desc": "Not equal to", "reverse": None, "series_examples": None, + "series_returns": _returns_series, + }, + "lt": { + "op": "<", + "desc": "Less than", + "reverse": None, + "series_examples": None, + "series_returns": _returns_series, }, - "lt": {"op": "<", "desc": "Less than", "reverse": None, "series_examples": None}, "le": { "op": "<=", "desc": "Less than or equal to", "reverse": None, "series_examples": None, + "series_returns": _returns_series, + }, + "gt": { + "op": ">", + "desc": "Greater than", + "reverse": None, + "series_examples": None, + "series_returns": _returns_series, }, - "gt": {"op": ">", "desc": "Greater than", "reverse": None, "series_examples": None}, "ge": { "op": ">=", "desc": "Greater than or equal to", "reverse": None, "series_examples": None, + "series_returns": _returns_series, }, } @@ -339,8 +373,7 @@ def _make_flex_doc(op_name, typ): Returns ------- -Series - The result of the operation. +{series_returns} See Also --------