maxframe 2.4.0rc1__cp312-cp312-win32.whl

maxframe/dataframe/arithmetic/mod.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 DataFrameMod(DataFrameBinopUfunc):
+    _op_type_ = opcodes.MOD
+
+    _func_name = "mod"
+    _rfunc_name = "rmod"
+
+    @classproperty
+    def _operator(self):
+        return operator.mod
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorMod
+
+        return TensorMod
+
+
+_mod_example = """
+>>> a.mod(b, fill_value=0).execute()
+a    0.0
+b    NaN
+c    NaN
+d    0.0
+e    NaN
+dtype: float64
+"""
+
+
+@bin_arithmetic_doc("Modulo", equiv="%", series_example=_mod_example)
+def mod(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameMod(axis=axis, level=level, fill_value=fill_value, lhs=df, rhs=other)
+    return op(df, other)
+
+
+@bin_arithmetic_doc("Modulo", equiv="%", series_example=_mod_example)
+def rmod(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameMod(axis=axis, level=level, fill_value=fill_value, lhs=other, rhs=df)
+    return op.rcall(df, other)

maxframe/dataframe/arithmetic/multiply.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 DataFrameMul(DataFrameBinopUfunc):
+    _op_type_ = opcodes.MUL
+
+    _func_name = "mul"
+    _rfunc_name = "rmul"
+
+    @classproperty
+    def _operator(self):
+        return operator.mul
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorMultiply
+
+        return TensorMultiply
+
+
+_mul_example = """
+>>> a.multiply(b, fill_value=0).execute()
+a    1.0
+b    0.0
+c    0.0
+d    0.0
+e    NaN
+dtype: float64
+"""
+
+
+@bin_arithmetic_doc("Multiplication", equiv="*", series_example=_mul_example)
+def mul(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameMul(axis=axis, level=level, fill_value=fill_value, lhs=df, rhs=other)
+    return op(df, other)
+
+
+@bin_arithmetic_doc("Multiplication", equiv="*", series_example=_mul_example)
+def rmul(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameMul(axis=axis, level=level, fill_value=fill_value, lhs=other, rhs=df)
+    return op.rcall(df, other)

maxframe/dataframe/arithmetic/negative.py

@@ -0,0 +1,33 @@
+# 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 DataFrameNegative(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.NEGATIVE
+    _func_name = "negative"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorNegative
+
+        return TensorNegative
+
+
+def negative(df):
+    op = DataFrameNegative()
+    return op(df)

maxframe/dataframe/arithmetic/not_equal.py

@@ -0,0 +1,58 @@
+# 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 numpy as np
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameBinopUfunc
+from .docstring import bin_compare_doc
+
+
+class DataFrameNotEqual(DataFrameBinopUfunc):
+    _op_type_ = opcodes.NE
+
+    _func_name = "ne"
+    _rfunc_name = "ne"
+
+    return_dtype = np.dtype(bool)
+
+    @classproperty
+    def _operator(self):
+        return lambda lhs, rhs: lhs.ne(rhs)
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorNotEqual
+
+        return TensorNotEqual
+
+
+_ne_example = """
+>>> a.ne(b, fill_value=0).execute()
+a    False
+b     True
+c     True
+d     True
+e     True
+dtype: bool
+"""
+
+
+@bin_compare_doc("Not equal to", equiv="!=", series_example=_ne_example)
+def ne(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameNotEqual(
+        axis=axis, level=level, lhs=df, rhs=other, fill_value=fill_value
+    )
+    return op(df, other)

maxframe/dataframe/arithmetic/power.py

@@ -0,0 +1,68 @@
+# 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 DataFramePower(DataFrameBinopUfunc):
+    _op_type_ = opcodes.POW
+
+    _func_name = "pow"
+    _rfunc_name = "rpow"
+
+    @classproperty
+    def _operator(self):
+        return operator.pow
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorPower
+
+        return TensorPower
+
+
+_pow_example = """
+>>> a.pow(b, fill_value=0).execute()
+a    1.0
+b    1.0
+c    1.0
+d    0.0
+e    NaN
+dtype: float64
+"""
+
+
+@bin_arithmetic_doc(
+    "Exponential power", op_name="pow", equiv="**", series_example=_pow_example
+)
+def power(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFramePower(
+        axis=axis, level=level, fill_value=fill_value, lhs=df, rhs=other
+    )
+    return op(df, other)
+
+
+@bin_arithmetic_doc(
+    "Exponential power", op_name="rpow", equiv="**", series_example=_pow_example
+)
+def rpower(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFramePower(
+        axis=axis, level=level, fill_value=fill_value, lhs=other, rhs=df
+    )
+    return op.rcall(df, other)

maxframe/dataframe/arithmetic/radians.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 DataFrameRadians(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.RADIANS
+    _func_name = "radians"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorRadians
+
+        return TensorRadians

maxframe/dataframe/arithmetic/round.py

@@ -0,0 +1,144 @@
+# 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 ...serialization.serializables import Int32Field
+from ...utils import classproperty
+from .core import DataFrameUnaryUfunc
+
+
+class DataFrameRound(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.AROUND
+    _func_name = "round"
+    _legacy_name = "DataFrameAround"  # since v2.3.0
+
+    decimals = Int32Field("decimals", default=None)
+
+    def __init__(self, output_types=None, **kw):
+        super().__init__(output_types=output_types, **kw)
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorAround
+
+        return TensorAround
+
+
+def round(df, decimals=0, *args, **kwargs):
+    if len(args) > 0:
+        raise TypeError(
+            f"round() takes 0 positional arguments but {len(args)} was given"
+        )
+    op = DataFrameRound(decimals=decimals, **kwargs)
+    return op(df)
+
+
+# keep for import compatibility
+DataFrameAround = DataFrameRound
+
+
+round.__frame_doc__ = """
+Round a DataFrame to a variable number of decimal places.
+
+Parameters
+----------
+decimals : int, dict
+    Number of decimal places to round each column to. If an int is
+    given, round each column to the same number of places.
+    Otherwise dict and Series round to variable numbers of places.
+    Column names should be in the keys if `decimals` is a
+    dict-like. Any columns not included in `decimals` will be left
+    as is. Elements of `decimals` which are not columns of the
+    input will be ignored.
+*args
+    Additional keywords have no effect but might be accepted for
+    compatibility with numpy.
+**kwargs
+    Additional keywords have no effect but might be accepted for
+    compatibility with numpy.
+
+Returns
+-------
+DataFrame
+    A DataFrame with the affected columns rounded to the specified
+    number of decimal places.
+
+See Also
+--------
+numpy.around : Round a numpy array to the given number of decimals.
+Series.round : Round a Series to the given number of decimals.
+
+Examples
+--------
+>>> import maxframe.dataframe as md
+>>> df = md.DataFrame([(.21, .32), (.01, .67), (.66, .03), (.21, .18)],
+...                   columns=['dogs', 'cats'])
+>>> df.execute()
+    dogs  cats
+0  0.21  0.32
+1  0.01  0.67
+2  0.66  0.03
+3  0.21  0.18
+
+By providing an integer each column is rounded to the same number
+of decimal places
+
+>>> df.round(1).execute()
+    dogs  cats
+0   0.2   0.3
+1   0.0   0.7
+2   0.7   0.0
+3   0.2   0.2
+
+With a dict, the number of places for specific columns can be
+specified with the column names as key and the number of decimal
+places as value
+
+>>> df.round({'dogs': 1, 'cats': 0}).execute()
+    dogs  cats
+0   0.2   0.0
+1   0.0   1.0
+2   0.7   0.0
+3   0.2   0.0
+"""
+round.__series_doc__ = """
+Round each value in a Series to the given number of decimals.
+
+Parameters
+----------
+decimals : int, default 0
+    Number of decimal places to round to. If decimals is negative,
+    it specifies the number of positions to the left of the decimal point.
+
+Returns
+-------
+Series
+    Rounded values of the Series.
+
+See Also
+--------
+numpy.around : Round values of an np.array.
+DataFrame.round : Round values of a DataFrame.
+
+Examples
+--------
+>>> import maxframe.tensor as mt
+>>> import maxframe.dataframe as md
+>>> s = md.Series([0.1, 1.3, 2.7])
+>>> s.round().execute()
+0    0.0
+1    1.0
+2    3.0
+dtype: float64
+"""

maxframe/dataframe/arithmetic/sin.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 DataFrameSin(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.SIN
+    _func_name = "sin"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorSin
+
+        return TensorSin

maxframe/dataframe/arithmetic/sinh.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 DataFrameSinh(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.SINH
+    _func_name = "sinh"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorSinh
+
+        return TensorSinh

maxframe/dataframe/arithmetic/sqrt.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 DataFrameSqrt(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.SQRT
+    _func_name = "sqrt"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorSqrt
+
+        return TensorSqrt

maxframe/dataframe/arithmetic/subtract.py

@@ -0,0 +1,64 @@
+# 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 DataFrameSubtract(DataFrameBinopUfunc):
+    _op_type_ = opcodes.SUB
+
+    _func_name = "sub"
+    _rfunc_name = "rsub"
+
+    @classproperty
+    def _operator(self):
+        return operator.sub
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorSubtract
+
+        return TensorSubtract
+
+
+_sub_example = """
+>>> a.subtract(b, fill_value=0).execute()
+a    0.0
+b    1.0
+c    1.0
+d   -1.0
+e    NaN
+dtype: float64
+"""
+
+
+@bin_arithmetic_doc("Subtraction", equiv="-", series_example=_sub_example)
+def subtract(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameSubtract(
+        axis=axis, level=level, fill_value=fill_value, lhs=df, rhs=other
+    )
+    return op(df, other)
+
+
+@bin_arithmetic_doc("Subtraction", equiv="-", series_example=_sub_example)
+def rsubtract(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameSubtract(
+        axis=axis, level=level, fill_value=fill_value, lhs=other, rhs=df
+    )
+    return op.rcall(df, other)

maxframe/dataframe/arithmetic/tan.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 DataFrameTan(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.TAN
+    _func_name = "tan"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorTan
+
+        return TensorTan

maxframe/dataframe/arithmetic/tanh.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 DataFrameTanh(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.TANH
+    _func_name = "tanh"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorTanh
+
+        return TensorTanh

maxframe/dataframe/arithmetic/tests/__init__.py

@@ -0,0 +1,13 @@
+# 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.