maxframe 2.4.0rc1__cp312-cp312-win32.whl

maxframe/dataframe/arithmetic/add.py

@@ -0,0 +1,60 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import operator
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameBinopUfunc
+from .docstring import bin_arithmetic_doc
+
+
+class DataFrameAdd(DataFrameBinopUfunc):
+    _op_type_ = opcodes.ADD
+
+    _func_name = "add"
+    _rfunc_name = "radd"
+
+    @classproperty
+    def _operator(self):
+        return operator.add
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorAdd
+
+        return TensorAdd
+
+
+_add_example = """
+>>> a.add(b, fill_value=0).execute()
+a    2.0
+b    1.0
+c    1.0
+d    1.0
+e    NaN
+dtype: float64
+"""
+
+
+@bin_arithmetic_doc("Addition", equiv="+", series_example=_add_example)
+def add(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameAdd(axis=axis, level=level, fill_value=fill_value, lhs=df, rhs=other)
+    return op(df, other)
+
+
+@bin_arithmetic_doc("Addition", equiv="+", series_example=_add_example)
+def radd(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameAdd(axis=axis, level=level, fill_value=fill_value, lhs=other, rhs=df)
+    return op.rcall(df, other)

maxframe/dataframe/arithmetic/arccos.py

@@ -0,0 +1,28 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameUnaryUfunc
+
+
+class DataFrameArccos(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.ARCCOS
+    _func_name = "arccos"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorArccos
+
+        return TensorArccos

maxframe/dataframe/arithmetic/arccosh.py

@@ -0,0 +1,28 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameUnaryUfunc
+
+
+class DataFrameArccosh(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.ARCCOSH
+    _func_name = "arccosh"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorArccosh
+
+        return TensorArccosh

maxframe/dataframe/arithmetic/arcsin.py

@@ -0,0 +1,28 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameUnaryUfunc
+
+
+class DataFrameArcsin(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.ARCSIN
+    _func_name = "arcsin"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorArcsin
+
+        return TensorArcsin

maxframe/dataframe/arithmetic/arcsinh.py

@@ -0,0 +1,28 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameUnaryUfunc
+
+
+class DataFrameArcsinh(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.ARCSINH
+    _func_name = "arcsinh"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorArcsinh
+
+        return TensorArcsinh

maxframe/dataframe/arithmetic/arctan.py

@@ -0,0 +1,28 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameUnaryUfunc
+
+
+class DataFrameArctan(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.ARCTAN
+    _func_name = "arctan"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorArctan
+
+        return TensorArctan

maxframe/dataframe/arithmetic/arctanh.py

@@ -0,0 +1,28 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameUnaryUfunc
+
+
+class DataFrameArctanh(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.ARCTANH
+    _func_name = "arctanh"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorArctanh
+
+        return TensorArctanh

maxframe/dataframe/arithmetic/between.py

@@ -0,0 +1,106 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+def between(series, left, right, inclusive="both"):
+    """
+    Return boolean Series equivalent to left <= series <= right.
+    This function returns a boolean vector containing `True` wherever the
+    corresponding Series element is between the boundary values `left` and
+    `right`. NA values are treated as `False`.
+
+    Parameters
+    ----------
+    left : scalar or list-like
+        Left boundary.
+    right : scalar or list-like
+        Right boundary.
+    inclusive : {"both", "neither", "left", "right"}
+        Include boundaries. Whether to set each bound as closed or open.
+
+    Returns
+    -------
+    Series
+        Series representing whether each element is between left and
+        right (inclusive).
+
+    See Also
+    --------
+    Series.gt : Greater than of series and other.
+    Series.lt : Less than of series and other.
+
+    Notes
+    -----
+    This function is equivalent to ``(left <= ser) & (ser <= right)``
+
+    Examples
+    --------
+    >>> import maxframe.dataframe as md
+    >>> s = md.Series([2, 0, 4, 8, np.nan])
+
+    Boundary values are included by default:
+
+    >>> s.between(1, 4).execute()
+    0     True
+    1    False
+    2     True
+    3    False
+    4    False
+    dtype: bool
+
+    With `inclusive` set to ``"neither"`` boundary values are excluded:
+
+    >>> s.between(1, 4, inclusive="neither").execute()
+    0     True
+    1    False
+    2    False
+    3    False
+    4    False
+    dtype: bool
+
+    `left` and `right` can be any scalar value:
+
+    >>> s = md.Series(['Alice', 'Bob', 'Carol', 'Eve'])
+    >>> s.between('Anna', 'Daniel').execute()
+    0    False
+    1     True
+    2     True
+    3    False
+    dtype: bool
+    """
+    if isinstance(inclusive, bool):  # pragma: no cover
+        # for pandas < 1.3.0
+        if inclusive:
+            inclusive = "both"
+        else:
+            inclusive = "neither"
+    if inclusive == "both":
+        lmask = series >= left
+        rmask = series <= right
+    elif inclusive == "left":
+        lmask = series >= left
+        rmask = series < right
+    elif inclusive == "right":
+        lmask = series > left
+        rmask = series <= right
+    elif inclusive == "neither":
+        lmask = series > left
+        rmask = series < right
+    else:
+        raise ValueError(
+            "Inclusive has to be either string of 'both',"
+            "'left', 'right', or 'neither'."
+        )
+
+    return lmask & rmask

maxframe/dataframe/arithmetic/bitwise_and.py

@@ -0,0 +1,46 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import operator
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameBinopUfunc
+
+
+class DataFrameAnd(DataFrameBinopUfunc):
+    _op_type_ = opcodes.AND
+
+    _bit_func_name = "__and__"
+    _bit_rfunc_name = "__rand__"
+
+    @classproperty
+    def _operator(self):
+        return operator.and_
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorBitand
+
+        return TensorBitand
+
+
+def bitand(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameAnd(axis=axis, level=level, fill_value=fill_value, lhs=df, rhs=other)
+    return op(df, other)
+
+
+def rbitand(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameAnd(axis=axis, level=level, fill_value=fill_value, lhs=other, rhs=df)
+    return op.rcall(df, other)

maxframe/dataframe/arithmetic/bitwise_or.py

@@ -0,0 +1,50 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import operator
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameArithmeticTreeMixin, DataFrameBinopUfunc
+
+
+class DataFrameOr(DataFrameBinopUfunc):
+    _op_type_ = opcodes.OR
+
+    _bit_func_name = "__or__"
+    _bit_rfunc_name = "__ror__"
+
+    @classproperty
+    def _operator(self):
+        return operator.or_
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorBitor
+
+        return TensorBitor
+
+
+class DataFrameTreeOr(DataFrameArithmeticTreeMixin, DataFrameOr):
+    _op_type_ = opcodes.TREE_OR
+
+
+def bitor(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameOr(axis=axis, level=level, fill_value=fill_value, lhs=df, rhs=other)
+    return op(df, other)
+
+
+def rbitor(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameOr(axis=axis, level=level, fill_value=fill_value, lhs=other, rhs=df)
+    return op.rcall(df, other)

maxframe/dataframe/arithmetic/bitwise_xor.py

@@ -0,0 +1,46 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import operator
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameBinopUfunc
+
+
+class DataFrameXor(DataFrameBinopUfunc):
+    _op_type_ = opcodes.XOR
+
+    _bit_func_name = "__xor__"
+    _bit_rfunc_name = "__rxor__"
+
+    @classproperty
+    def _operator(self):
+        return operator.xor
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorBitxor
+
+        return TensorBitxor
+
+
+def bitxor(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameXor(axis=axis, level=level, fill_value=fill_value, lhs=df, rhs=other)
+    return op(df, other)
+
+
+def rbitxor(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameXor(axis=axis, level=level, fill_value=fill_value, lhs=other, rhs=df)
+    return op.rcall(df, other)

maxframe/dataframe/arithmetic/ceil.py

@@ -0,0 +1,28 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameUnaryUfunc
+
+
+class DataFrameCeil(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.CEIL
+    _func_name = "ceil"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorCeil
+
+        return TensorCeil