maxframe 0.1.0b5__cp311-cp311-macosx_11_0_arm64.whl

maxframe/dataframe/arithmetic/arcsinh.py

@@ -0,0 +1,28 @@
+# Copyright 1999-2024 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-2024 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-2024 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/around.py

@@ -0,0 +1,152 @@
+# Copyright 1999-2024 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 DataFrameAround(DataFrameUnaryUfunc):
+    _op_type_ = opcodes.AROUND
+    _func_name = "around"
+
+    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 around(df, decimals=0, *args, **kwargs):
+    if len(args) > 0:
+        raise TypeError(
+            f"round() takes 0 positional arguments but {len(args)} was given"
+        )
+    op = DataFrameAround(decimals=decimals, **kwargs)
+    return op(df)
+
+
+around.__frame_doc__ = """
+Round a DataFrame to a variable number of decimal places.
+
+Parameters
+----------
+decimals : int, dict, Series
+    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, or in the index if `decimals` is a Series. 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
+
+Using a Series, the number of places for specific columns can be
+specified with the column names as index and the number of
+decimal places as value
+
+>>> decimals = md.Series([0, 1], index=['cats', 'dogs'])
+>>> df.round(decimals).execute()
+    dogs  cats
+0   0.2   0.0
+1   0.0   1.0
+2   0.7   0.0
+3   0.2   0.0
+"""
+around.__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/bitwise_and.py

@@ -0,0 +1,46 @@
+# Copyright 1999-2024 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-2024 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-2024 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-2024 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