maxframe 2.4.0rc1__cp312-cp312-win32.whl

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

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

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

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

maxframe/dataframe/arithmetic/floordiv.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 DataFrameFloorDiv(DataFrameBinopUfunc):
+    _op_type_ = opcodes.FLOORDIV
+
+    _func_name = "floordiv"
+    _rfunc_name = "rfloordiv"
+
+    @classproperty
+    def _operator(self):
+        return operator.floordiv
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorFloorDiv
+
+        return TensorFloorDiv
+
+
+_floordiv_example = """
+>>> a.floordiv(b, fill_value=0).execute()
+a    1.0
+b    NaN
+c    NaN
+d    0.0
+e    NaN
+dtype: float64
+"""
+
+
+@bin_arithmetic_doc("Integer division", equiv="//", series_example=_floordiv_example)
+def floordiv(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameFloorDiv(
+        axis=axis, level=level, fill_value=fill_value, lhs=df, rhs=other
+    )
+    return op(df, other)
+
+
+@bin_arithmetic_doc("Integer division", equiv="//", series_example=_floordiv_example)
+def rfloordiv(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameFloorDiv(
+        axis=axis, level=level, fill_value=fill_value, lhs=other, rhs=df
+    )
+    return op.rcall(df, other)

maxframe/dataframe/arithmetic/greater.py

@@ -0,0 +1,59 @@
+# 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 DataFrameGreater(DataFrameBinopUfunc):
+    _op_type_ = opcodes.GT
+
+    _func_name = "gt"
+    _rfunc_name = "lt"
+
+    return_dtype = np.dtype(bool)
+
+    @classproperty
+    def _operator(self):
+        return lambda lhs, rhs: lhs.gt(rhs)
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorGreaterThan
+
+        return TensorGreaterThan
+
+
+_gt_example = """
+>>> a.gt(b, fill_value=0).execute()
+a     True
+b    False
+c    False
+d    False
+e     True
+f    False
+dtype: bool
+"""
+
+
+@bin_compare_doc("Greater than", equiv=">", series_example=_gt_example)
+def gt(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameGreater(
+        axis=axis, level=level, lhs=df, rhs=other, fill_value=fill_value
+    )
+    return op(df, other)

maxframe/dataframe/arithmetic/greater_equal.py

@@ -0,0 +1,59 @@
+# 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 DataFrameGreaterEqual(DataFrameBinopUfunc):
+    _op_type_ = opcodes.GE
+
+    _func_name = "ge"
+    _rfunc_name = "le"
+
+    return_dtype = np.dtype(bool)
+
+    @classproperty
+    def _operator(self):
+        return lambda lhs, rhs: lhs.ge(rhs)
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorGreaterEqual
+
+        return TensorGreaterEqual
+
+
+_ge_example = """
+>>> a.ge(b, fill_value=0).execute()
+a     True
+b     True
+c    False
+d    False
+e     True
+f    False
+dtype: bool
+"""
+
+
+@bin_compare_doc("Greater than or equal to", equiv=">=", series_example=_ge_example)
+def ge(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameGreaterEqual(
+        axis=axis, level=level, lhs=df, rhs=other, fill_value=fill_value
+    )
+    return op(df, other)

maxframe/dataframe/arithmetic/invert.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 DataFrameNot(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.INVERT
+    _func_name = "__invert__"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorNot
+
+        return TensorNot
+
+
+def invert(df):
+    op = DataFrameNot()
+    return op(df)

maxframe/dataframe/arithmetic/is_ufuncs.py

@@ -0,0 +1,62 @@
+# 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
+import pandas as pd
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameUnaryUfunc
+
+
+class DataFrameIsUFuncMixin:
+    @classmethod
+    def _get_output_dtype(cls, df):
+        if df.ndim == 2:
+            return pd.Series(np.dtype(bool), index=df.dtypes.index)
+        else:
+            return np.dtype(bool)
+
+
+class DataFrameIsNan(DataFrameIsUFuncMixin, DataFrameUnaryUfunc):
+    _op_type_ = opcodes.ISNAN
+    _func_name = "isnan"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorIsNan
+
+        return TensorIsNan
+
+
+class DataFrameIsInf(DataFrameIsUFuncMixin, DataFrameUnaryUfunc):
+    _op_type_ = opcodes.ISINF
+    _func_name = "isinf"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorIsInf
+
+        return TensorIsInf
+
+
+class DataFrameIsFinite(DataFrameIsUFuncMixin, DataFrameUnaryUfunc):
+    _op_type_ = opcodes.ISFINITE
+    _func_name = "isfinite"
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorIsFinite
+
+        return TensorIsFinite

maxframe/dataframe/arithmetic/less.py

@@ -0,0 +1,57 @@
+# 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 DataFrameLess(DataFrameBinopUfunc):
+    _op_type_ = opcodes.LT
+
+    _func_name = "lt"
+    _rfunc_name = "gt"
+
+    return_dtype = np.dtype(bool)
+
+    @classproperty
+    def _operator(self):
+        return lambda lhs, rhs: lhs.lt(rhs)
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorLessThan
+
+        return TensorLessThan
+
+
+_lt_example = """
+>>> a.lt(b, fill_value=0).execute()
+a    False
+b    False
+c     True
+d    False
+e    False
+f     True
+dtype: bool
+"""
+
+
+@bin_compare_doc("Less than", equiv="<", series_example=_lt_example)
+def lt(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameLess(axis=axis, level=level, lhs=df, rhs=other, fill_value=fill_value)
+    return op(df, other)

maxframe/dataframe/arithmetic/less_equal.py

@@ -0,0 +1,59 @@
+# 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 DataFrameLessEqual(DataFrameBinopUfunc):
+    _op_type_ = opcodes.LE
+
+    _func_name = "le"
+    _rfunc_name = "ge"
+
+    return_dtype = np.dtype(bool)
+
+    @classproperty
+    def _operator(self):
+        return lambda lhs, rhs: lhs.le(rhs)
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorLessEqual
+
+        return TensorLessEqual
+
+
+_le_example = """
+>>> a.le(b, fill_value=0).execute()
+a    False
+b     True
+c     True
+d    False
+e    False
+f     True
+dtype: bool
+"""
+
+
+@bin_compare_doc("Less than or equal to", equiv="<=", series_example=_le_example)
+def le(df, other, axis="columns", level=None, fill_value=None):
+    op = DataFrameLessEqual(
+        axis=axis, level=level, lhs=df, rhs=other, fill_value=fill_value
+    )
+    return op(df, other)

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

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

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

maxframe/dataframe/arithmetic/maximum.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.
+
+import numpy as np
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameBinopUfunc
+
+
+class DataFrameMaximum(DataFrameBinopUfunc):
+    _op_type_ = opcodes.MAXIMUM
+
+    @classproperty
+    def _operator(self):
+        return np.maximum
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorMaximum
+
+        return TensorMaximum

maxframe/dataframe/arithmetic/minimum.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.
+
+import numpy as np
+
+from ... import opcodes
+from ...utils import classproperty
+from .core import DataFrameBinopUfunc
+
+
+class DataFrameMinimum(DataFrameBinopUfunc):
+    _op_type_ = opcodes.MINIMUM
+
+    @classproperty
+    def _operator(self):
+        return np.minimum
+
+    @classproperty
+    def tensor_op_type(self):
+        from ...tensor.arithmetic import TensorMinimum
+
+        return TensorMinimum