maxframe 2.4.0rc1__cp312-cp312-win32.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- maxframe/__init__.py +33 -0
- maxframe/_utils.cp312-win32.pyd +0 -0
- maxframe/_utils.pxd +33 -0
- maxframe/_utils.pyi +21 -0
- maxframe/_utils.pyx +561 -0
- maxframe/codegen/__init__.py +27 -0
- maxframe/codegen/core.py +597 -0
- maxframe/codegen/spe/__init__.py +16 -0
- maxframe/codegen/spe/core.py +307 -0
- maxframe/codegen/spe/dataframe/__init__.py +38 -0
- maxframe/codegen/spe/dataframe/accessors/__init__.py +15 -0
- maxframe/codegen/spe/dataframe/accessors/base.py +71 -0
- maxframe/codegen/spe/dataframe/accessors/dict_.py +89 -0
- maxframe/codegen/spe/dataframe/accessors/list_.py +44 -0
- maxframe/codegen/spe/dataframe/accessors/struct_.py +28 -0
- maxframe/codegen/spe/dataframe/arithmetic.py +89 -0
- maxframe/codegen/spe/dataframe/datasource.py +181 -0
- maxframe/codegen/spe/dataframe/datastore.py +204 -0
- maxframe/codegen/spe/dataframe/extensions.py +63 -0
- maxframe/codegen/spe/dataframe/fetch.py +26 -0
- maxframe/codegen/spe/dataframe/groupby.py +312 -0
- maxframe/codegen/spe/dataframe/indexing.py +333 -0
- maxframe/codegen/spe/dataframe/merge.py +110 -0
- maxframe/codegen/spe/dataframe/misc.py +264 -0
- maxframe/codegen/spe/dataframe/missing.py +64 -0
- maxframe/codegen/spe/dataframe/reduction.py +183 -0
- maxframe/codegen/spe/dataframe/reshape.py +46 -0
- maxframe/codegen/spe/dataframe/sort.py +104 -0
- maxframe/codegen/spe/dataframe/statistics.py +46 -0
- maxframe/codegen/spe/dataframe/tests/__init__.py +13 -0
- maxframe/codegen/spe/dataframe/tests/accessors/__init__.py +13 -0
- maxframe/codegen/spe/dataframe/tests/accessors/test_base.py +33 -0
- maxframe/codegen/spe/dataframe/tests/accessors/test_dict.py +304 -0
- maxframe/codegen/spe/dataframe/tests/accessors/test_list.py +134 -0
- maxframe/codegen/spe/dataframe/tests/accessors/test_struct.py +75 -0
- maxframe/codegen/spe/dataframe/tests/indexing/__init__.py +13 -0
- maxframe/codegen/spe/dataframe/tests/indexing/conftest.py +58 -0
- maxframe/codegen/spe/dataframe/tests/indexing/test_getitem.py +124 -0
- maxframe/codegen/spe/dataframe/tests/indexing/test_iloc.py +95 -0
- maxframe/codegen/spe/dataframe/tests/indexing/test_indexing.py +39 -0
- maxframe/codegen/spe/dataframe/tests/indexing/test_loc.py +35 -0
- maxframe/codegen/spe/dataframe/tests/indexing/test_rename.py +51 -0
- maxframe/codegen/spe/dataframe/tests/indexing/test_reset_index.py +88 -0
- maxframe/codegen/spe/dataframe/tests/indexing/test_sample.py +45 -0
- maxframe/codegen/spe/dataframe/tests/indexing/test_set_axis.py +45 -0
- maxframe/codegen/spe/dataframe/tests/indexing/test_set_index.py +41 -0
- maxframe/codegen/spe/dataframe/tests/indexing/test_setitem.py +46 -0
- maxframe/codegen/spe/dataframe/tests/misc/__init__.py +13 -0
- maxframe/codegen/spe/dataframe/tests/misc/test_apply.py +133 -0
- maxframe/codegen/spe/dataframe/tests/misc/test_drop_duplicates.py +92 -0
- maxframe/codegen/spe/dataframe/tests/misc/test_misc.py +202 -0
- maxframe/codegen/spe/dataframe/tests/missing/__init__.py +13 -0
- maxframe/codegen/spe/dataframe/tests/missing/test_checkna.py +94 -0
- maxframe/codegen/spe/dataframe/tests/missing/test_dropna.py +50 -0
- maxframe/codegen/spe/dataframe/tests/missing/test_fillna.py +94 -0
- maxframe/codegen/spe/dataframe/tests/missing/test_replace.py +45 -0
- maxframe/codegen/spe/dataframe/tests/test_arithmetic.py +73 -0
- maxframe/codegen/spe/dataframe/tests/test_datasource.py +184 -0
- maxframe/codegen/spe/dataframe/tests/test_datastore.py +200 -0
- maxframe/codegen/spe/dataframe/tests/test_extensions.py +88 -0
- maxframe/codegen/spe/dataframe/tests/test_groupby.py +288 -0
- maxframe/codegen/spe/dataframe/tests/test_merge.py +426 -0
- maxframe/codegen/spe/dataframe/tests/test_reduction.py +117 -0
- maxframe/codegen/spe/dataframe/tests/test_reshape.py +79 -0
- maxframe/codegen/spe/dataframe/tests/test_sort.py +179 -0
- maxframe/codegen/spe/dataframe/tests/test_statistics.py +70 -0
- maxframe/codegen/spe/dataframe/tests/test_tseries.py +29 -0
- maxframe/codegen/spe/dataframe/tests/test_value_counts.py +60 -0
- maxframe/codegen/spe/dataframe/tests/test_window.py +69 -0
- maxframe/codegen/spe/dataframe/tseries.py +55 -0
- maxframe/codegen/spe/dataframe/udf.py +62 -0
- maxframe/codegen/spe/dataframe/value_counts.py +31 -0
- maxframe/codegen/spe/dataframe/window.py +65 -0
- maxframe/codegen/spe/learn/__init__.py +15 -0
- maxframe/codegen/spe/learn/contrib/__init__.py +15 -0
- maxframe/codegen/spe/learn/contrib/lightgbm.py +161 -0
- maxframe/codegen/spe/learn/contrib/models.py +41 -0
- maxframe/codegen/spe/learn/contrib/pytorch.py +49 -0
- maxframe/codegen/spe/learn/contrib/tests/__init__.py +13 -0
- maxframe/codegen/spe/learn/contrib/tests/test_lightgbm.py +123 -0
- maxframe/codegen/spe/learn/contrib/tests/test_models.py +41 -0
- maxframe/codegen/spe/learn/contrib/tests/test_pytorch.py +53 -0
- maxframe/codegen/spe/learn/contrib/tests/test_xgboost.py +99 -0
- maxframe/codegen/spe/learn/contrib/xgboost.py +152 -0
- maxframe/codegen/spe/learn/metrics/__init__.py +15 -0
- maxframe/codegen/spe/learn/metrics/_classification.py +120 -0
- maxframe/codegen/spe/learn/metrics/_ranking.py +76 -0
- maxframe/codegen/spe/learn/metrics/pairwise.py +51 -0
- maxframe/codegen/spe/learn/metrics/tests/__init__.py +13 -0
- maxframe/codegen/spe/learn/metrics/tests/test_classification.py +93 -0
- maxframe/codegen/spe/learn/metrics/tests/test_pairwise.py +36 -0
- maxframe/codegen/spe/learn/metrics/tests/test_ranking.py +59 -0
- maxframe/codegen/spe/learn/model_selection/__init__.py +13 -0
- maxframe/codegen/spe/learn/model_selection/tests/__init__.py +13 -0
- maxframe/codegen/spe/learn/model_selection/tests/test_split.py +41 -0
- maxframe/codegen/spe/learn/preprocessing/__init__.py +15 -0
- maxframe/codegen/spe/learn/preprocessing/_data.py +37 -0
- maxframe/codegen/spe/learn/preprocessing/_label.py +47 -0
- maxframe/codegen/spe/learn/preprocessing/tests/__init__.py +13 -0
- maxframe/codegen/spe/learn/preprocessing/tests/test_data.py +31 -0
- maxframe/codegen/spe/learn/preprocessing/tests/test_label.py +43 -0
- maxframe/codegen/spe/learn/utils/__init__.py +15 -0
- maxframe/codegen/spe/learn/utils/checks.py +55 -0
- maxframe/codegen/spe/learn/utils/multiclass.py +60 -0
- maxframe/codegen/spe/learn/utils/shuffle.py +85 -0
- maxframe/codegen/spe/learn/utils/sparsefuncs.py +35 -0
- maxframe/codegen/spe/learn/utils/tests/__init__.py +13 -0
- maxframe/codegen/spe/learn/utils/tests/test_checks.py +48 -0
- maxframe/codegen/spe/learn/utils/tests/test_multiclass.py +52 -0
- maxframe/codegen/spe/learn/utils/tests/test_shuffle.py +50 -0
- maxframe/codegen/spe/learn/utils/tests/test_sparsefuncs.py +34 -0
- maxframe/codegen/spe/learn/utils/tests/test_validation.py +44 -0
- maxframe/codegen/spe/learn/utils/validation.py +35 -0
- maxframe/codegen/spe/objects.py +26 -0
- maxframe/codegen/spe/remote.py +29 -0
- maxframe/codegen/spe/tensor/__init__.py +31 -0
- maxframe/codegen/spe/tensor/arithmetic.py +95 -0
- maxframe/codegen/spe/tensor/core.py +41 -0
- maxframe/codegen/spe/tensor/datasource.py +166 -0
- maxframe/codegen/spe/tensor/extensions.py +35 -0
- maxframe/codegen/spe/tensor/fetch.py +26 -0
- maxframe/codegen/spe/tensor/fft.py +74 -0
- maxframe/codegen/spe/tensor/indexing.py +63 -0
- maxframe/codegen/spe/tensor/linalg.py +90 -0
- maxframe/codegen/spe/tensor/merge.py +31 -0
- maxframe/codegen/spe/tensor/misc.py +175 -0
- maxframe/codegen/spe/tensor/random.py +29 -0
- maxframe/codegen/spe/tensor/reduction.py +39 -0
- maxframe/codegen/spe/tensor/reshape.py +26 -0
- maxframe/codegen/spe/tensor/sort.py +42 -0
- maxframe/codegen/spe/tensor/spatial.py +45 -0
- maxframe/codegen/spe/tensor/special.py +35 -0
- maxframe/codegen/spe/tensor/statistics.py +68 -0
- maxframe/codegen/spe/tensor/tests/__init__.py +13 -0
- maxframe/codegen/spe/tensor/tests/test_arithmetic.py +103 -0
- maxframe/codegen/spe/tensor/tests/test_datasource.py +99 -0
- maxframe/codegen/spe/tensor/tests/test_extensions.py +37 -0
- maxframe/codegen/spe/tensor/tests/test_fft.py +64 -0
- maxframe/codegen/spe/tensor/tests/test_indexing.py +44 -0
- maxframe/codegen/spe/tensor/tests/test_linalg.py +52 -0
- maxframe/codegen/spe/tensor/tests/test_merge.py +28 -0
- maxframe/codegen/spe/tensor/tests/test_misc.py +144 -0
- maxframe/codegen/spe/tensor/tests/test_random.py +55 -0
- maxframe/codegen/spe/tensor/tests/test_reduction.py +65 -0
- maxframe/codegen/spe/tensor/tests/test_reshape.py +39 -0
- maxframe/codegen/spe/tensor/tests/test_sort.py +49 -0
- maxframe/codegen/spe/tensor/tests/test_spatial.py +33 -0
- maxframe/codegen/spe/tensor/tests/test_special.py +28 -0
- maxframe/codegen/spe/tensor/tests/test_statistics.py +43 -0
- maxframe/codegen/spe/tests/__init__.py +13 -0
- maxframe/codegen/spe/tests/test_remote.py +29 -0
- maxframe/codegen/spe/tests/test_spe_codegen.py +135 -0
- maxframe/codegen/spe/utils.py +56 -0
- maxframe/codegen/tests/__init__.py +13 -0
- maxframe/codegen/tests/test_codegen.py +67 -0
- maxframe/config/__init__.py +15 -0
- maxframe/config/config.py +630 -0
- maxframe/config/tests/__init__.py +13 -0
- maxframe/config/tests/test_config.py +114 -0
- maxframe/config/tests/test_validators.py +46 -0
- maxframe/config/validators.py +142 -0
- maxframe/conftest.py +261 -0
- maxframe/core/__init__.py +53 -0
- maxframe/core/accessor.py +45 -0
- maxframe/core/base.py +157 -0
- maxframe/core/context.py +110 -0
- maxframe/core/entity/__init__.py +34 -0
- maxframe/core/entity/core.py +150 -0
- maxframe/core/entity/executable.py +337 -0
- maxframe/core/entity/objects.py +115 -0
- maxframe/core/entity/output_types.py +101 -0
- maxframe/core/entity/tests/__init__.py +13 -0
- maxframe/core/entity/tests/test_objects.py +42 -0
- maxframe/core/entity/tileables.py +376 -0
- maxframe/core/entity/utils.py +39 -0
- maxframe/core/graph/__init__.py +22 -0
- maxframe/core/graph/builder/__init__.py +15 -0
- maxframe/core/graph/builder/base.py +90 -0
- maxframe/core/graph/builder/tileable.py +34 -0
- maxframe/core/graph/builder/utils.py +37 -0
- maxframe/core/graph/core.cp312-win32.pyd +0 -0
- maxframe/core/graph/core.pyx +478 -0
- maxframe/core/graph/entity.py +187 -0
- maxframe/core/graph/tests/__init__.py +13 -0
- maxframe/core/graph/tests/test_graph.py +205 -0
- maxframe/core/mode.py +101 -0
- maxframe/core/operator/__init__.py +32 -0
- maxframe/core/operator/base.py +481 -0
- maxframe/core/operator/core.py +307 -0
- maxframe/core/operator/fetch.py +40 -0
- maxframe/core/operator/objects.py +43 -0
- maxframe/core/operator/shuffle.py +45 -0
- maxframe/core/operator/tests/__init__.py +13 -0
- maxframe/core/operator/tests/test_core.py +64 -0
- maxframe/core/operator/utils.py +68 -0
- maxframe/core/tests/__init__.py +13 -0
- maxframe/core/tests/test_mode.py +75 -0
- maxframe/dataframe/__init__.py +90 -0
- maxframe/dataframe/accessors/__init__.py +20 -0
- maxframe/dataframe/accessors/compat.py +45 -0
- maxframe/dataframe/accessors/datetime_/__init__.py +35 -0
- maxframe/dataframe/accessors/datetime_/accessor.py +67 -0
- maxframe/dataframe/accessors/datetime_/core.py +106 -0
- maxframe/dataframe/accessors/datetime_/tests/__init__.py +13 -0
- maxframe/dataframe/accessors/datetime_/tests/test_datetime_accessor.py +41 -0
- maxframe/dataframe/accessors/dict_/__init__.py +45 -0
- maxframe/dataframe/accessors/dict_/accessor.py +39 -0
- maxframe/dataframe/accessors/dict_/contains.py +72 -0
- maxframe/dataframe/accessors/dict_/core.py +48 -0
- maxframe/dataframe/accessors/dict_/getitem.py +140 -0
- maxframe/dataframe/accessors/dict_/length.py +64 -0
- maxframe/dataframe/accessors/dict_/remove.py +75 -0
- maxframe/dataframe/accessors/dict_/setitem.py +79 -0
- maxframe/dataframe/accessors/dict_/tests/__init__.py +13 -0
- maxframe/dataframe/accessors/dict_/tests/test_dict_accessor.py +168 -0
- maxframe/dataframe/accessors/list_/__init__.py +39 -0
- maxframe/dataframe/accessors/list_/accessor.py +39 -0
- maxframe/dataframe/accessors/list_/core.py +48 -0
- maxframe/dataframe/accessors/list_/getitem.py +128 -0
- maxframe/dataframe/accessors/list_/length.py +64 -0
- maxframe/dataframe/accessors/list_/tests/__init__.py +13 -0
- maxframe/dataframe/accessors/list_/tests/test_list_accessor.py +81 -0
- maxframe/dataframe/accessors/plotting/__init__.py +40 -0
- maxframe/dataframe/accessors/plotting/core.py +78 -0
- maxframe/dataframe/accessors/plotting/tests/__init__.py +13 -0
- maxframe/dataframe/accessors/plotting/tests/test_plotting_accessor.py +136 -0
- maxframe/dataframe/accessors/string_/__init__.py +36 -0
- maxframe/dataframe/accessors/string_/accessor.py +215 -0
- maxframe/dataframe/accessors/string_/core.py +226 -0
- maxframe/dataframe/accessors/string_/tests/__init__.py +13 -0
- maxframe/dataframe/accessors/string_/tests/test_string_accessor.py +73 -0
- maxframe/dataframe/accessors/struct_/__init__.py +39 -0
- maxframe/dataframe/accessors/struct_/accessor.py +39 -0
- maxframe/dataframe/accessors/struct_/core.py +43 -0
- maxframe/dataframe/accessors/struct_/dtypes.py +53 -0
- maxframe/dataframe/accessors/struct_/field.py +123 -0
- maxframe/dataframe/accessors/struct_/tests/__init__.py +13 -0
- maxframe/dataframe/accessors/struct_/tests/test_struct_accessor.py +91 -0
- maxframe/dataframe/arithmetic/__init__.py +373 -0
- maxframe/dataframe/arithmetic/abs.py +33 -0
- maxframe/dataframe/arithmetic/add.py +60 -0
- maxframe/dataframe/arithmetic/arccos.py +28 -0
- maxframe/dataframe/arithmetic/arccosh.py +28 -0
- maxframe/dataframe/arithmetic/arcsin.py +28 -0
- maxframe/dataframe/arithmetic/arcsinh.py +28 -0
- maxframe/dataframe/arithmetic/arctan.py +28 -0
- maxframe/dataframe/arithmetic/arctanh.py +28 -0
- maxframe/dataframe/arithmetic/between.py +106 -0
- maxframe/dataframe/arithmetic/bitwise_and.py +46 -0
- maxframe/dataframe/arithmetic/bitwise_or.py +50 -0
- maxframe/dataframe/arithmetic/bitwise_xor.py +46 -0
- maxframe/dataframe/arithmetic/ceil.py +28 -0
- maxframe/dataframe/arithmetic/core.py +361 -0
- maxframe/dataframe/arithmetic/cos.py +28 -0
- maxframe/dataframe/arithmetic/cosh.py +28 -0
- maxframe/dataframe/arithmetic/degrees.py +28 -0
- maxframe/dataframe/arithmetic/docstring.py +416 -0
- maxframe/dataframe/arithmetic/dot.py +237 -0
- maxframe/dataframe/arithmetic/equal.py +58 -0
- maxframe/dataframe/arithmetic/exp.py +28 -0
- maxframe/dataframe/arithmetic/exp2.py +28 -0
- maxframe/dataframe/arithmetic/expm1.py +28 -0
- maxframe/dataframe/arithmetic/floor.py +28 -0
- maxframe/dataframe/arithmetic/floordiv.py +64 -0
- maxframe/dataframe/arithmetic/greater.py +59 -0
- maxframe/dataframe/arithmetic/greater_equal.py +59 -0
- maxframe/dataframe/arithmetic/invert.py +33 -0
- maxframe/dataframe/arithmetic/is_ufuncs.py +62 -0
- maxframe/dataframe/arithmetic/less.py +57 -0
- maxframe/dataframe/arithmetic/less_equal.py +59 -0
- maxframe/dataframe/arithmetic/log.py +28 -0
- maxframe/dataframe/arithmetic/log10.py +28 -0
- maxframe/dataframe/arithmetic/log2.py +28 -0
- maxframe/dataframe/arithmetic/maximum.py +33 -0
- maxframe/dataframe/arithmetic/minimum.py +33 -0
- maxframe/dataframe/arithmetic/mod.py +60 -0
- maxframe/dataframe/arithmetic/multiply.py +60 -0
- maxframe/dataframe/arithmetic/negative.py +33 -0
- maxframe/dataframe/arithmetic/not_equal.py +58 -0
- maxframe/dataframe/arithmetic/power.py +68 -0
- maxframe/dataframe/arithmetic/radians.py +28 -0
- maxframe/dataframe/arithmetic/round.py +144 -0
- maxframe/dataframe/arithmetic/sin.py +28 -0
- maxframe/dataframe/arithmetic/sinh.py +28 -0
- maxframe/dataframe/arithmetic/sqrt.py +28 -0
- maxframe/dataframe/arithmetic/subtract.py +64 -0
- maxframe/dataframe/arithmetic/tan.py +28 -0
- maxframe/dataframe/arithmetic/tanh.py +28 -0
- maxframe/dataframe/arithmetic/tests/__init__.py +13 -0
- maxframe/dataframe/arithmetic/tests/test_arithmetic.py +747 -0
- maxframe/dataframe/arithmetic/truediv.py +64 -0
- maxframe/dataframe/arithmetic/trunc.py +28 -0
- maxframe/dataframe/core.py +2386 -0
- maxframe/dataframe/datasource/__init__.py +33 -0
- maxframe/dataframe/datasource/core.py +112 -0
- maxframe/dataframe/datasource/dataframe.py +59 -0
- maxframe/dataframe/datasource/date_range.py +512 -0
- maxframe/dataframe/datasource/direct.py +57 -0
- maxframe/dataframe/datasource/from_dict.py +124 -0
- maxframe/dataframe/datasource/from_index.py +58 -0
- maxframe/dataframe/datasource/from_records.py +191 -0
- maxframe/dataframe/datasource/from_tensor.py +503 -0
- maxframe/dataframe/datasource/index.py +117 -0
- maxframe/dataframe/datasource/read_csv.py +534 -0
- maxframe/dataframe/datasource/read_odps_query.py +536 -0
- maxframe/dataframe/datasource/read_odps_table.py +295 -0
- maxframe/dataframe/datasource/read_parquet.py +278 -0
- maxframe/dataframe/datasource/series.py +55 -0
- maxframe/dataframe/datasource/tests/__init__.py +13 -0
- maxframe/dataframe/datasource/tests/test_datasource.py +663 -0
- maxframe/dataframe/datastore/__init__.py +41 -0
- maxframe/dataframe/datastore/core.py +28 -0
- maxframe/dataframe/datastore/direct.py +268 -0
- maxframe/dataframe/datastore/tests/__init__.py +13 -0
- maxframe/dataframe/datastore/tests/test_to_odps.py +99 -0
- maxframe/dataframe/datastore/to_csv.py +219 -0
- maxframe/dataframe/datastore/to_json.py +215 -0
- maxframe/dataframe/datastore/to_odps.py +285 -0
- maxframe/dataframe/datastore/to_parquet.py +121 -0
- maxframe/dataframe/extensions/__init__.py +70 -0
- maxframe/dataframe/extensions/accessor.py +35 -0
- maxframe/dataframe/extensions/apply_chunk.py +733 -0
- maxframe/dataframe/extensions/cartesian_chunk.py +153 -0
- maxframe/dataframe/extensions/collect_kv.py +126 -0
- maxframe/dataframe/extensions/extract_kv.py +177 -0
- maxframe/dataframe/extensions/flatjson.py +133 -0
- maxframe/dataframe/extensions/flatmap.py +329 -0
- maxframe/dataframe/extensions/map_reduce.py +263 -0
- maxframe/dataframe/extensions/rebalance.py +62 -0
- maxframe/dataframe/extensions/reshuffle.py +83 -0
- maxframe/dataframe/extensions/tests/__init__.py +13 -0
- maxframe/dataframe/extensions/tests/test_apply_chunk.py +194 -0
- maxframe/dataframe/extensions/tests/test_extensions.py +198 -0
- maxframe/dataframe/extensions/tests/test_map_reduce.py +135 -0
- maxframe/dataframe/fetch/__init__.py +15 -0
- maxframe/dataframe/fetch/core.py +97 -0
- maxframe/dataframe/groupby/__init__.py +105 -0
- maxframe/dataframe/groupby/aggregation.py +485 -0
- maxframe/dataframe/groupby/apply.py +235 -0
- maxframe/dataframe/groupby/apply_chunk.py +407 -0
- maxframe/dataframe/groupby/core.py +342 -0
- maxframe/dataframe/groupby/cum.py +102 -0
- maxframe/dataframe/groupby/expanding.py +264 -0
- maxframe/dataframe/groupby/extensions.py +26 -0
- maxframe/dataframe/groupby/fill.py +149 -0
- maxframe/dataframe/groupby/getitem.py +105 -0
- maxframe/dataframe/groupby/head.py +115 -0
- maxframe/dataframe/groupby/rank.py +136 -0
- maxframe/dataframe/groupby/rolling.py +206 -0
- maxframe/dataframe/groupby/sample.py +214 -0
- maxframe/dataframe/groupby/shift.py +114 -0
- maxframe/dataframe/groupby/tests/__init__.py +13 -0
- maxframe/dataframe/groupby/tests/test_groupby.py +373 -0
- maxframe/dataframe/groupby/transform.py +264 -0
- maxframe/dataframe/indexing/__init__.py +104 -0
- maxframe/dataframe/indexing/add_prefix_suffix.py +110 -0
- maxframe/dataframe/indexing/align.py +350 -0
- maxframe/dataframe/indexing/at.py +83 -0
- maxframe/dataframe/indexing/droplevel.py +195 -0
- maxframe/dataframe/indexing/filter.py +169 -0
- maxframe/dataframe/indexing/get_level_values.py +76 -0
- maxframe/dataframe/indexing/getitem.py +205 -0
- maxframe/dataframe/indexing/iat.py +82 -0
- maxframe/dataframe/indexing/iloc.py +711 -0
- maxframe/dataframe/indexing/insert.py +118 -0
- maxframe/dataframe/indexing/loc.py +694 -0
- maxframe/dataframe/indexing/reindex.py +541 -0
- maxframe/dataframe/indexing/rename.py +445 -0
- maxframe/dataframe/indexing/rename_axis.py +217 -0
- maxframe/dataframe/indexing/reorder_levels.py +143 -0
- maxframe/dataframe/indexing/reset_index.py +427 -0
- maxframe/dataframe/indexing/sample.py +232 -0
- maxframe/dataframe/indexing/set_axis.py +197 -0
- maxframe/dataframe/indexing/set_index.py +128 -0
- maxframe/dataframe/indexing/setitem.py +133 -0
- maxframe/dataframe/indexing/swaplevel.py +185 -0
- maxframe/dataframe/indexing/take.py +99 -0
- maxframe/dataframe/indexing/tests/__init__.py +13 -0
- maxframe/dataframe/indexing/tests/test_indexing.py +488 -0
- maxframe/dataframe/indexing/truncate.py +140 -0
- maxframe/dataframe/indexing/where.py +300 -0
- maxframe/dataframe/indexing/xs.py +148 -0
- maxframe/dataframe/initializer.py +298 -0
- maxframe/dataframe/merge/__init__.py +53 -0
- maxframe/dataframe/merge/append.py +120 -0
- maxframe/dataframe/merge/combine.py +244 -0
- maxframe/dataframe/merge/combine_first.py +120 -0
- maxframe/dataframe/merge/compare.py +387 -0
- maxframe/dataframe/merge/concat.py +500 -0
- maxframe/dataframe/merge/merge.py +806 -0
- maxframe/dataframe/merge/tests/__init__.py +13 -0
- maxframe/dataframe/merge/tests/test_merge.py +390 -0
- maxframe/dataframe/merge/update.py +271 -0
- maxframe/dataframe/misc/__init__.py +145 -0
- maxframe/dataframe/misc/_duplicate.py +56 -0
- maxframe/dataframe/misc/apply.py +730 -0
- maxframe/dataframe/misc/astype.py +237 -0
- maxframe/dataframe/misc/case_when.py +145 -0
- maxframe/dataframe/misc/check_monotonic.py +84 -0
- maxframe/dataframe/misc/check_unique.py +82 -0
- maxframe/dataframe/misc/clip.py +145 -0
- maxframe/dataframe/misc/cut.py +386 -0
- maxframe/dataframe/misc/describe.py +278 -0
- maxframe/dataframe/misc/diff.py +210 -0
- maxframe/dataframe/misc/drop.py +473 -0
- maxframe/dataframe/misc/drop_duplicates.py +251 -0
- maxframe/dataframe/misc/duplicated.py +292 -0
- maxframe/dataframe/misc/eval.py +730 -0
- maxframe/dataframe/misc/explode.py +171 -0
- maxframe/dataframe/misc/factorize.py +160 -0
- maxframe/dataframe/misc/get_dummies.py +241 -0
- maxframe/dataframe/misc/infer_dtypes.py +251 -0
- maxframe/dataframe/misc/isin.py +220 -0
- maxframe/dataframe/misc/map.py +360 -0
- maxframe/dataframe/misc/memory_usage.py +248 -0
- maxframe/dataframe/misc/pct_change.py +68 -0
- maxframe/dataframe/misc/qcut.py +104 -0
- maxframe/dataframe/misc/rechunk.py +59 -0
- maxframe/dataframe/misc/repeat.py +159 -0
- maxframe/dataframe/misc/select_dtypes.py +104 -0
- maxframe/dataframe/misc/shift.py +259 -0
- maxframe/dataframe/misc/tests/__init__.py +13 -0
- maxframe/dataframe/misc/tests/test_misc.py +649 -0
- maxframe/dataframe/misc/to_numeric.py +181 -0
- maxframe/dataframe/misc/transform.py +346 -0
- maxframe/dataframe/misc/transpose.py +148 -0
- maxframe/dataframe/misc/valid_index.py +115 -0
- maxframe/dataframe/misc/value_counts.py +206 -0
- maxframe/dataframe/missing/__init__.py +53 -0
- maxframe/dataframe/missing/checkna.py +231 -0
- maxframe/dataframe/missing/dropna.py +294 -0
- maxframe/dataframe/missing/fillna.py +283 -0
- maxframe/dataframe/missing/replace.py +446 -0
- maxframe/dataframe/missing/tests/__init__.py +13 -0
- maxframe/dataframe/missing/tests/test_missing.py +90 -0
- maxframe/dataframe/operators.py +231 -0
- maxframe/dataframe/reduction/__init__.py +129 -0
- maxframe/dataframe/reduction/aggregation.py +502 -0
- maxframe/dataframe/reduction/all.py +78 -0
- maxframe/dataframe/reduction/any.py +78 -0
- maxframe/dataframe/reduction/argmax.py +103 -0
- maxframe/dataframe/reduction/argmin.py +103 -0
- maxframe/dataframe/reduction/core.py +923 -0
- maxframe/dataframe/reduction/count.py +63 -0
- maxframe/dataframe/reduction/cov.py +166 -0
- maxframe/dataframe/reduction/cummax.py +30 -0
- maxframe/dataframe/reduction/cummin.py +30 -0
- maxframe/dataframe/reduction/cumprod.py +30 -0
- maxframe/dataframe/reduction/cumsum.py +30 -0
- maxframe/dataframe/reduction/custom_reduction.py +42 -0
- maxframe/dataframe/reduction/idxmax.py +185 -0
- maxframe/dataframe/reduction/idxmin.py +185 -0
- maxframe/dataframe/reduction/kurtosis.py +111 -0
- maxframe/dataframe/reduction/max.py +65 -0
- maxframe/dataframe/reduction/mean.py +63 -0
- maxframe/dataframe/reduction/median.py +56 -0
- maxframe/dataframe/reduction/min.py +65 -0
- maxframe/dataframe/reduction/mode.py +190 -0
- maxframe/dataframe/reduction/nunique.py +149 -0
- maxframe/dataframe/reduction/prod.py +81 -0
- maxframe/dataframe/reduction/reduction_size.py +36 -0
- maxframe/dataframe/reduction/sem.py +73 -0
- maxframe/dataframe/reduction/skew.py +93 -0
- maxframe/dataframe/reduction/std.py +53 -0
- maxframe/dataframe/reduction/str_concat.py +51 -0
- maxframe/dataframe/reduction/sum.py +81 -0
- maxframe/dataframe/reduction/tests/__init__.py +13 -0
- maxframe/dataframe/reduction/tests/test_reduction.py +598 -0
- maxframe/dataframe/reduction/unique.py +153 -0
- maxframe/dataframe/reduction/var.py +76 -0
- maxframe/dataframe/reshape/__init__.py +38 -0
- maxframe/dataframe/reshape/melt.py +169 -0
- maxframe/dataframe/reshape/pivot.py +233 -0
- maxframe/dataframe/reshape/pivot_table.py +275 -0
- maxframe/dataframe/reshape/stack.py +240 -0
- maxframe/dataframe/reshape/unstack.py +114 -0
- maxframe/dataframe/sort/__init__.py +49 -0
- maxframe/dataframe/sort/argsort.py +68 -0
- maxframe/dataframe/sort/core.py +37 -0
- maxframe/dataframe/sort/nlargest.py +238 -0
- maxframe/dataframe/sort/nsmallest.py +228 -0
- maxframe/dataframe/sort/rank.py +147 -0
- maxframe/dataframe/sort/sort_index.py +153 -0
- maxframe/dataframe/sort/sort_values.py +308 -0
- maxframe/dataframe/sort/tests/__init__.py +13 -0
- maxframe/dataframe/sort/tests/test_sort.py +85 -0
- maxframe/dataframe/statistics/__init__.py +33 -0
- maxframe/dataframe/statistics/corr.py +284 -0
- maxframe/dataframe/statistics/quantile.py +338 -0
- maxframe/dataframe/statistics/tests/__init__.py +13 -0
- maxframe/dataframe/statistics/tests/test_statistics.py +82 -0
- maxframe/dataframe/tests/__init__.py +13 -0
- maxframe/dataframe/tests/test_initializer.py +60 -0
- maxframe/dataframe/tests/test_typing.py +119 -0
- maxframe/dataframe/tests/test_utils.py +169 -0
- maxframe/dataframe/tseries/__init__.py +32 -0
- maxframe/dataframe/tseries/at_time.py +61 -0
- maxframe/dataframe/tseries/between_time.py +122 -0
- maxframe/dataframe/tseries/tests/__init__.py +13 -0
- maxframe/dataframe/tseries/tests/test_tseries.py +30 -0
- maxframe/dataframe/tseries/to_datetime.py +299 -0
- maxframe/dataframe/typing_.py +196 -0
- maxframe/dataframe/ufunc/__init__.py +27 -0
- maxframe/dataframe/ufunc/tensor.py +54 -0
- maxframe/dataframe/ufunc/ufunc.py +53 -0
- maxframe/dataframe/utils.py +1728 -0
- maxframe/dataframe/window/__init__.py +29 -0
- maxframe/dataframe/window/aggregation.py +100 -0
- maxframe/dataframe/window/core.py +82 -0
- maxframe/dataframe/window/ewm.py +247 -0
- maxframe/dataframe/window/expanding.py +151 -0
- maxframe/dataframe/window/rolling.py +389 -0
- maxframe/dataframe/window/tests/__init__.py +13 -0
- maxframe/dataframe/window/tests/test_ewm.py +70 -0
- maxframe/dataframe/window/tests/test_expanding.py +60 -0
- maxframe/dataframe/window/tests/test_rolling.py +57 -0
- maxframe/env.py +37 -0
- maxframe/errors.py +52 -0
- maxframe/extension.py +131 -0
- maxframe/io/__init__.py +13 -0
- maxframe/io/objects/__init__.py +24 -0
- maxframe/io/objects/core.py +156 -0
- maxframe/io/objects/tensor.py +133 -0
- maxframe/io/objects/tests/__init__.py +13 -0
- maxframe/io/objects/tests/test_object_io.py +85 -0
- maxframe/io/odpsio/__init__.py +24 -0
- maxframe/io/odpsio/arrow.py +161 -0
- maxframe/io/odpsio/schema.py +533 -0
- maxframe/io/odpsio/tableio.py +736 -0
- maxframe/io/odpsio/tests/__init__.py +13 -0
- maxframe/io/odpsio/tests/test_arrow.py +132 -0
- maxframe/io/odpsio/tests/test_schema.py +582 -0
- maxframe/io/odpsio/tests/test_tableio.py +205 -0
- maxframe/io/odpsio/tests/test_volumeio.py +75 -0
- maxframe/io/odpsio/volumeio.py +102 -0
- maxframe/learn/__init__.py +25 -0
- maxframe/learn/cluster/__init__.py +15 -0
- maxframe/learn/cluster/_kmeans.py +782 -0
- maxframe/learn/contrib/__init__.py +17 -0
- maxframe/learn/contrib/graph/__init__.py +15 -0
- maxframe/learn/contrib/graph/connected_components.py +216 -0
- maxframe/learn/contrib/graph/tests/__init__.py +13 -0
- maxframe/learn/contrib/graph/tests/test_connected_components.py +53 -0
- maxframe/learn/contrib/lightgbm/__init__.py +33 -0
- maxframe/learn/contrib/lightgbm/_predict.py +138 -0
- maxframe/learn/contrib/lightgbm/_train.py +163 -0
- maxframe/learn/contrib/lightgbm/callback.py +114 -0
- maxframe/learn/contrib/lightgbm/classifier.py +199 -0
- maxframe/learn/contrib/lightgbm/core.py +372 -0
- maxframe/learn/contrib/lightgbm/dataset.py +153 -0
- maxframe/learn/contrib/lightgbm/regressor.py +29 -0
- maxframe/learn/contrib/lightgbm/tests/__init__.py +13 -0
- maxframe/learn/contrib/lightgbm/tests/test_callback.py +58 -0
- maxframe/learn/contrib/llm/__init__.py +17 -0
- maxframe/learn/contrib/llm/core.py +105 -0
- maxframe/learn/contrib/llm/deploy/__init__.py +13 -0
- maxframe/learn/contrib/llm/deploy/config.py +221 -0
- maxframe/learn/contrib/llm/deploy/core.py +247 -0
- maxframe/learn/contrib/llm/deploy/framework.py +35 -0
- maxframe/learn/contrib/llm/deploy/loader.py +360 -0
- maxframe/learn/contrib/llm/deploy/tests/__init__.py +13 -0
- maxframe/learn/contrib/llm/deploy/tests/test_register_models.py +359 -0
- maxframe/learn/contrib/llm/models/__init__.py +16 -0
- maxframe/learn/contrib/llm/models/dashscope.py +114 -0
- maxframe/learn/contrib/llm/models/managed.py +119 -0
- maxframe/learn/contrib/llm/models/openai.py +72 -0
- maxframe/learn/contrib/llm/multi_modal.py +135 -0
- maxframe/learn/contrib/llm/tests/__init__.py +13 -0
- maxframe/learn/contrib/llm/tests/test_core.py +34 -0
- maxframe/learn/contrib/llm/tests/test_openai.py +187 -0
- maxframe/learn/contrib/llm/tests/test_text_gen.py +155 -0
- maxframe/learn/contrib/llm/text.py +608 -0
- maxframe/learn/contrib/models.py +109 -0
- maxframe/learn/contrib/pytorch/__init__.py +16 -0
- maxframe/learn/contrib/pytorch/run_function.py +110 -0
- maxframe/learn/contrib/pytorch/run_script.py +102 -0
- maxframe/learn/contrib/pytorch/tests/__init__.py +13 -0
- maxframe/learn/contrib/pytorch/tests/test_pytorch.py +42 -0
- maxframe/learn/contrib/utils.py +108 -0
- maxframe/learn/contrib/xgboost/__init__.py +33 -0
- maxframe/learn/contrib/xgboost/callback.py +86 -0
- maxframe/learn/contrib/xgboost/classifier.py +119 -0
- maxframe/learn/contrib/xgboost/core.py +469 -0
- maxframe/learn/contrib/xgboost/dmatrix.py +157 -0
- maxframe/learn/contrib/xgboost/predict.py +133 -0
- maxframe/learn/contrib/xgboost/regressor.py +91 -0
- maxframe/learn/contrib/xgboost/tests/__init__.py +13 -0
- maxframe/learn/contrib/xgboost/tests/test_callback.py +41 -0
- maxframe/learn/contrib/xgboost/tests/test_core.py +43 -0
- maxframe/learn/contrib/xgboost/train.py +181 -0
- maxframe/learn/core.py +344 -0
- maxframe/learn/datasets/__init__.py +20 -0
- maxframe/learn/datasets/samples_generator.py +628 -0
- maxframe/learn/linear_model/__init__.py +15 -0
- maxframe/learn/linear_model/_base.py +220 -0
- maxframe/learn/linear_model/_lin_reg.py +175 -0
- maxframe/learn/metrics/__init__.py +31 -0
- maxframe/learn/metrics/_check_targets.py +95 -0
- maxframe/learn/metrics/_classification.py +1266 -0
- maxframe/learn/metrics/_ranking.py +477 -0
- maxframe/learn/metrics/_regression.py +256 -0
- maxframe/learn/metrics/_scorer.py +60 -0
- maxframe/learn/metrics/pairwise/__init__.py +21 -0
- maxframe/learn/metrics/pairwise/core.py +77 -0
- maxframe/learn/metrics/pairwise/cosine.py +115 -0
- maxframe/learn/metrics/pairwise/euclidean.py +176 -0
- maxframe/learn/metrics/pairwise/haversine.py +96 -0
- maxframe/learn/metrics/pairwise/manhattan.py +80 -0
- maxframe/learn/metrics/pairwise/pairwise.py +127 -0
- maxframe/learn/metrics/pairwise/pairwise_distances_topk.py +121 -0
- maxframe/learn/metrics/pairwise/rbf_kernel.py +51 -0
- maxframe/learn/metrics/tests/__init__.py +13 -0
- maxframe/learn/metrics/tests/test_scorer.py +26 -0
- maxframe/learn/model_selection/__init__.py +15 -0
- maxframe/learn/model_selection/_split.py +451 -0
- maxframe/learn/model_selection/tests/__init__.py +13 -0
- maxframe/learn/model_selection/tests/test_split.py +156 -0
- maxframe/learn/preprocessing/__init__.py +16 -0
- maxframe/learn/preprocessing/_data/__init__.py +17 -0
- maxframe/learn/preprocessing/_data/min_max_scaler.py +401 -0
- maxframe/learn/preprocessing/_data/normalize.py +127 -0
- maxframe/learn/preprocessing/_data/standard_scaler.py +512 -0
- maxframe/learn/preprocessing/_data/utils.py +79 -0
- maxframe/learn/preprocessing/_label/__init__.py +16 -0
- maxframe/learn/preprocessing/_label/_label_binarizer.py +599 -0
- maxframe/learn/preprocessing/_label/_label_encoder.py +174 -0
- maxframe/learn/utils/__init__.py +20 -0
- maxframe/learn/utils/_encode.py +312 -0
- maxframe/learn/utils/checks.py +160 -0
- maxframe/learn/utils/core.py +121 -0
- maxframe/learn/utils/extmath.py +246 -0
- maxframe/learn/utils/multiclass.py +292 -0
- maxframe/learn/utils/odpsio.py +262 -0
- maxframe/learn/utils/shuffle.py +114 -0
- maxframe/learn/utils/sparsefuncs.py +87 -0
- maxframe/learn/utils/validation.py +775 -0
- maxframe/lib/__init__.py +13 -0
- maxframe/lib/aio/__init__.py +27 -0
- maxframe/lib/aio/_runners.py +162 -0
- maxframe/lib/aio/_threads.py +35 -0
- maxframe/lib/aio/base.py +82 -0
- maxframe/lib/aio/file.py +85 -0
- maxframe/lib/aio/isolation.py +100 -0
- maxframe/lib/aio/lru.py +242 -0
- maxframe/lib/aio/parallelism.py +37 -0
- maxframe/lib/aio/tests/__init__.py +13 -0
- maxframe/lib/aio/tests/test_aio_file.py +55 -0
- maxframe/lib/compat.py +185 -0
- maxframe/lib/compression.py +55 -0
- maxframe/lib/cython/__init__.py +13 -0
- maxframe/lib/cython/libcpp.pxd +30 -0
- maxframe/lib/dtypes_extension/__init__.py +30 -0
- maxframe/lib/dtypes_extension/_fake_arrow_dtype.py +609 -0
- maxframe/lib/dtypes_extension/blob.py +304 -0
- maxframe/lib/dtypes_extension/dtypes.py +106 -0
- maxframe/lib/dtypes_extension/tests/__init__.py +13 -0
- maxframe/lib/dtypes_extension/tests/test_blob.py +88 -0
- maxframe/lib/dtypes_extension/tests/test_dtypes.py +63 -0
- maxframe/lib/dtypes_extension/tests/test_fake_arrow_dtype.py +75 -0
- maxframe/lib/filesystem/__init__.py +22 -0
- maxframe/lib/filesystem/_glob.py +173 -0
- maxframe/lib/filesystem/_oss_lib/__init__.py +13 -0
- maxframe/lib/filesystem/_oss_lib/common.py +274 -0
- maxframe/lib/filesystem/_oss_lib/glob.py +147 -0
- maxframe/lib/filesystem/_oss_lib/handle.py +180 -0
- maxframe/lib/filesystem/arrow.py +240 -0
- maxframe/lib/filesystem/base.py +327 -0
- maxframe/lib/filesystem/core.py +95 -0
- maxframe/lib/filesystem/fshandler.py +136 -0
- maxframe/lib/filesystem/fsmap.py +164 -0
- maxframe/lib/filesystem/hdfs.py +31 -0
- maxframe/lib/filesystem/local.py +120 -0
- maxframe/lib/filesystem/oss.py +283 -0
- maxframe/lib/filesystem/tests/__init__.py +13 -0
- maxframe/lib/filesystem/tests/test_filesystem.py +205 -0
- maxframe/lib/filesystem/tests/test_fshandler.py +281 -0
- maxframe/lib/filesystem/tests/test_oss.py +220 -0
- maxframe/lib/functools_compat.py +81 -0
- maxframe/lib/mmh3.cp312-win32.pyd +0 -0
- maxframe/lib/mmh3.pyi +43 -0
- maxframe/lib/mmh3_src/MurmurHash3.cpp +339 -0
- maxframe/lib/mmh3_src/MurmurHash3.h +43 -0
- maxframe/lib/mmh3_src/mmh3module.cpp +387 -0
- maxframe/lib/sparse/__init__.py +856 -0
- maxframe/lib/sparse/array.py +1616 -0
- maxframe/lib/sparse/core.py +90 -0
- maxframe/lib/sparse/linalg.py +31 -0
- maxframe/lib/sparse/matrix.py +244 -0
- maxframe/lib/sparse/tests/__init__.py +13 -0
- maxframe/lib/sparse/tests/test_sparse.py +476 -0
- maxframe/lib/sparse/vector.py +148 -0
- maxframe/lib/tblib/LICENSE +20 -0
- maxframe/lib/tblib/__init__.py +327 -0
- maxframe/lib/tblib/cpython.py +83 -0
- maxframe/lib/tblib/decorators.py +44 -0
- maxframe/lib/tblib/pickling_support.py +90 -0
- maxframe/lib/tests/__init__.py +13 -0
- maxframe/lib/tests/test_wrapped_pickle.py +51 -0
- maxframe/lib/version.py +620 -0
- maxframe/lib/wrapped_pickle.py +177 -0
- maxframe/mixin.py +157 -0
- maxframe/opcodes.py +654 -0
- maxframe/protocol.py +611 -0
- maxframe/remote/__init__.py +18 -0
- maxframe/remote/core.py +212 -0
- maxframe/remote/run_script.py +124 -0
- maxframe/serialization/__init__.py +39 -0
- maxframe/serialization/arrow.py +107 -0
- maxframe/serialization/blob.py +32 -0
- maxframe/serialization/core.cp312-win32.pyd +0 -0
- maxframe/serialization/core.pxd +50 -0
- maxframe/serialization/core.pyi +66 -0
- maxframe/serialization/core.pyx +1282 -0
- maxframe/serialization/exception.py +90 -0
- maxframe/serialization/maxframe_objects.py +39 -0
- maxframe/serialization/numpy.py +110 -0
- maxframe/serialization/pandas.py +278 -0
- maxframe/serialization/scipy.py +71 -0
- maxframe/serialization/serializables/__init__.py +55 -0
- maxframe/serialization/serializables/core.py +469 -0
- maxframe/serialization/serializables/field.py +624 -0
- maxframe/serialization/serializables/field_type.py +592 -0
- maxframe/serialization/serializables/tests/__init__.py +13 -0
- maxframe/serialization/serializables/tests/test_field_type.py +119 -0
- maxframe/serialization/serializables/tests/test_serializable.py +313 -0
- maxframe/serialization/tests/__init__.py +13 -0
- maxframe/serialization/tests/test_serial.py +516 -0
- maxframe/session.py +1250 -0
- maxframe/sperunner.py +165 -0
- maxframe/tensor/__init__.py +325 -0
- maxframe/tensor/arithmetic/__init__.py +322 -0
- maxframe/tensor/arithmetic/abs.py +66 -0
- maxframe/tensor/arithmetic/absolute.py +66 -0
- maxframe/tensor/arithmetic/add.py +112 -0
- maxframe/tensor/arithmetic/angle.py +70 -0
- maxframe/tensor/arithmetic/arccos.py +101 -0
- maxframe/tensor/arithmetic/arccosh.py +89 -0
- maxframe/tensor/arithmetic/arcsin.py +92 -0
- maxframe/tensor/arithmetic/arcsinh.py +84 -0
- maxframe/tensor/arithmetic/arctan.py +104 -0
- maxframe/tensor/arithmetic/arctan2.py +126 -0
- maxframe/tensor/arithmetic/arctanh.py +84 -0
- maxframe/tensor/arithmetic/around.py +112 -0
- maxframe/tensor/arithmetic/bitand.py +93 -0
- maxframe/tensor/arithmetic/bitor.py +100 -0
- maxframe/tensor/arithmetic/bitxor.py +93 -0
- maxframe/tensor/arithmetic/cbrt.py +64 -0
- maxframe/tensor/arithmetic/ceil.py +69 -0
- maxframe/tensor/arithmetic/clip.py +165 -0
- maxframe/tensor/arithmetic/conj.py +72 -0
- maxframe/tensor/arithmetic/copysign.py +76 -0
- maxframe/tensor/arithmetic/core.py +546 -0
- maxframe/tensor/arithmetic/cos.py +83 -0
- maxframe/tensor/arithmetic/cosh.py +70 -0
- maxframe/tensor/arithmetic/deg2rad.py +70 -0
- maxframe/tensor/arithmetic/degrees.py +75 -0
- maxframe/tensor/arithmetic/divide.py +112 -0
- maxframe/tensor/arithmetic/equal.py +74 -0
- maxframe/tensor/arithmetic/exp.py +104 -0
- maxframe/tensor/arithmetic/exp2.py +65 -0
- maxframe/tensor/arithmetic/expm1.py +77 -0
- maxframe/tensor/arithmetic/fabs.py +72 -0
- maxframe/tensor/arithmetic/fix.py +67 -0
- maxframe/tensor/arithmetic/float_power.py +101 -0
- maxframe/tensor/arithmetic/floor.py +75 -0
- maxframe/tensor/arithmetic/floordiv.py +92 -0
- maxframe/tensor/arithmetic/fmax.py +103 -0
- maxframe/tensor/arithmetic/fmin.py +104 -0
- maxframe/tensor/arithmetic/fmod.py +97 -0
- maxframe/tensor/arithmetic/frexp.py +96 -0
- maxframe/tensor/arithmetic/greater.py +75 -0
- maxframe/tensor/arithmetic/greater_equal.py +67 -0
- maxframe/tensor/arithmetic/hypot.py +75 -0
- maxframe/tensor/arithmetic/i0.py +87 -0
- maxframe/tensor/arithmetic/imag.py +65 -0
- maxframe/tensor/arithmetic/invert.py +108 -0
- maxframe/tensor/arithmetic/isclose.py +114 -0
- maxframe/tensor/arithmetic/iscomplex.py +62 -0
- maxframe/tensor/arithmetic/iscomplexobj.py +53 -0
- maxframe/tensor/arithmetic/isfinite.py +104 -0
- maxframe/tensor/arithmetic/isinf.py +101 -0
- maxframe/tensor/arithmetic/isnan.py +80 -0
- maxframe/tensor/arithmetic/isreal.py +61 -0
- maxframe/tensor/arithmetic/ldexp.py +97 -0
- maxframe/tensor/arithmetic/less.py +67 -0
- maxframe/tensor/arithmetic/less_equal.py +67 -0
- maxframe/tensor/arithmetic/log.py +90 -0
- maxframe/tensor/arithmetic/log10.py +83 -0
- maxframe/tensor/arithmetic/log1p.py +93 -0
- maxframe/tensor/arithmetic/log2.py +83 -0
- maxframe/tensor/arithmetic/logaddexp.py +78 -0
- maxframe/tensor/arithmetic/logaddexp2.py +76 -0
- maxframe/tensor/arithmetic/logical_and.py +79 -0
- maxframe/tensor/arithmetic/logical_not.py +72 -0
- maxframe/tensor/arithmetic/logical_or.py +80 -0
- maxframe/tensor/arithmetic/logical_xor.py +86 -0
- maxframe/tensor/arithmetic/lshift.py +80 -0
- maxframe/tensor/arithmetic/maximum.py +106 -0
- maxframe/tensor/arithmetic/minimum.py +106 -0
- maxframe/tensor/arithmetic/mod.py +102 -0
- maxframe/tensor/arithmetic/modf.py +87 -0
- maxframe/tensor/arithmetic/multiply.py +114 -0
- maxframe/tensor/arithmetic/nan_to_num.py +97 -0
- maxframe/tensor/arithmetic/negative.py +63 -0
- maxframe/tensor/arithmetic/nextafter.py +66 -0
- maxframe/tensor/arithmetic/not_equal.py +70 -0
- maxframe/tensor/arithmetic/positive.py +45 -0
- maxframe/tensor/arithmetic/power.py +104 -0
- maxframe/tensor/arithmetic/rad2deg.py +69 -0
- maxframe/tensor/arithmetic/radians.py +75 -0
- maxframe/tensor/arithmetic/real.py +68 -0
- maxframe/tensor/arithmetic/reciprocal.py +78 -0
- maxframe/tensor/arithmetic/rint.py +66 -0
- maxframe/tensor/arithmetic/rshift.py +79 -0
- maxframe/tensor/arithmetic/setimag.py +27 -0
- maxframe/tensor/arithmetic/setreal.py +27 -0
- maxframe/tensor/arithmetic/sign.py +79 -0
- maxframe/tensor/arithmetic/signbit.py +63 -0
- maxframe/tensor/arithmetic/sin.py +96 -0
- maxframe/tensor/arithmetic/sinc.py +100 -0
- maxframe/tensor/arithmetic/sinh.py +91 -0
- maxframe/tensor/arithmetic/spacing.py +70 -0
- maxframe/tensor/arithmetic/sqrt.py +79 -0
- maxframe/tensor/arithmetic/square.py +67 -0
- maxframe/tensor/arithmetic/subtract.py +83 -0
- maxframe/tensor/arithmetic/tan.py +86 -0
- maxframe/tensor/arithmetic/tanh.py +90 -0
- maxframe/tensor/arithmetic/tests/__init__.py +13 -0
- maxframe/tensor/arithmetic/tests/test_arithmetic.py +449 -0
- maxframe/tensor/arithmetic/truediv.py +102 -0
- maxframe/tensor/arithmetic/trunc.py +70 -0
- maxframe/tensor/arithmetic/utils.py +91 -0
- maxframe/tensor/array_utils.py +164 -0
- maxframe/tensor/core.py +597 -0
- maxframe/tensor/datasource/__init__.py +40 -0
- maxframe/tensor/datasource/arange.py +154 -0
- maxframe/tensor/datasource/array.py +399 -0
- maxframe/tensor/datasource/core.py +114 -0
- maxframe/tensor/datasource/diag.py +140 -0
- maxframe/tensor/datasource/diagflat.py +69 -0
- maxframe/tensor/datasource/empty.py +167 -0
- maxframe/tensor/datasource/eye.py +95 -0
- maxframe/tensor/datasource/from_dataframe.py +68 -0
- maxframe/tensor/datasource/from_dense.py +37 -0
- maxframe/tensor/datasource/from_sparse.py +45 -0
- maxframe/tensor/datasource/full.py +184 -0
- maxframe/tensor/datasource/identity.py +54 -0
- maxframe/tensor/datasource/indices.py +115 -0
- maxframe/tensor/datasource/linspace.py +140 -0
- maxframe/tensor/datasource/meshgrid.py +135 -0
- maxframe/tensor/datasource/ones.py +178 -0
- maxframe/tensor/datasource/scalar.py +40 -0
- maxframe/tensor/datasource/tests/__init__.py +13 -0
- maxframe/tensor/datasource/tests/test_datasource.py +310 -0
- maxframe/tensor/datasource/tri_array.py +107 -0
- maxframe/tensor/datasource/zeros.py +192 -0
- maxframe/tensor/extensions/__init__.py +33 -0
- maxframe/tensor/extensions/accessor.py +25 -0
- maxframe/tensor/extensions/apply_chunk.py +137 -0
- maxframe/tensor/extensions/rebalance.py +65 -0
- maxframe/tensor/fetch/__init__.py +15 -0
- maxframe/tensor/fetch/core.py +54 -0
- maxframe/tensor/fft/__init__.py +32 -0
- maxframe/tensor/fft/core.py +168 -0
- maxframe/tensor/fft/fft.py +112 -0
- maxframe/tensor/fft/fft2.py +118 -0
- maxframe/tensor/fft/fftfreq.py +80 -0
- maxframe/tensor/fft/fftn.py +123 -0
- maxframe/tensor/fft/fftshift.py +79 -0
- maxframe/tensor/fft/hfft.py +112 -0
- maxframe/tensor/fft/ifft.py +114 -0
- maxframe/tensor/fft/ifft2.py +115 -0
- maxframe/tensor/fft/ifftn.py +123 -0
- maxframe/tensor/fft/ifftshift.py +73 -0
- maxframe/tensor/fft/ihfft.py +93 -0
- maxframe/tensor/fft/irfft.py +118 -0
- maxframe/tensor/fft/irfft2.py +62 -0
- maxframe/tensor/fft/irfftn.py +114 -0
- maxframe/tensor/fft/rfft.py +116 -0
- maxframe/tensor/fft/rfft2.py +63 -0
- maxframe/tensor/fft/rfftfreq.py +87 -0
- maxframe/tensor/fft/rfftn.py +113 -0
- maxframe/tensor/indexing/__init__.py +47 -0
- maxframe/tensor/indexing/choose.py +198 -0
- maxframe/tensor/indexing/compress.py +122 -0
- maxframe/tensor/indexing/core.py +190 -0
- maxframe/tensor/indexing/extract.py +69 -0
- maxframe/tensor/indexing/fill_diagonal.py +180 -0
- maxframe/tensor/indexing/flatnonzero.py +58 -0
- maxframe/tensor/indexing/getitem.py +144 -0
- maxframe/tensor/indexing/nonzero.py +118 -0
- maxframe/tensor/indexing/setitem.py +142 -0
- maxframe/tensor/indexing/slice.py +32 -0
- maxframe/tensor/indexing/take.py +128 -0
- maxframe/tensor/indexing/tests/__init__.py +13 -0
- maxframe/tensor/indexing/tests/test_indexing.py +232 -0
- maxframe/tensor/indexing/unravel_index.py +103 -0
- maxframe/tensor/lib/__init__.py +16 -0
- maxframe/tensor/lib/index_tricks.py +404 -0
- maxframe/tensor/linalg/__init__.py +43 -0
- maxframe/tensor/linalg/_einsumfunc.py +1025 -0
- maxframe/tensor/linalg/cholesky.py +117 -0
- maxframe/tensor/linalg/dot.py +145 -0
- maxframe/tensor/linalg/einsum.py +339 -0
- maxframe/tensor/linalg/inner.py +36 -0
- maxframe/tensor/linalg/inv.py +83 -0
- maxframe/tensor/linalg/lstsq.py +100 -0
- maxframe/tensor/linalg/lu.py +115 -0
- maxframe/tensor/linalg/matmul.py +225 -0
- maxframe/tensor/linalg/matrix_norm.py +75 -0
- maxframe/tensor/linalg/norm.py +249 -0
- maxframe/tensor/linalg/qr.py +124 -0
- maxframe/tensor/linalg/solve.py +72 -0
- maxframe/tensor/linalg/solve_triangular.py +103 -0
- maxframe/tensor/linalg/svd.py +167 -0
- maxframe/tensor/linalg/tensordot.py +213 -0
- maxframe/tensor/linalg/vdot.py +73 -0
- maxframe/tensor/linalg/vector_norm.py +113 -0
- maxframe/tensor/merge/__init__.py +21 -0
- maxframe/tensor/merge/append.py +74 -0
- maxframe/tensor/merge/column_stack.py +63 -0
- maxframe/tensor/merge/concatenate.py +103 -0
- maxframe/tensor/merge/dstack.py +71 -0
- maxframe/tensor/merge/hstack.py +70 -0
- maxframe/tensor/merge/stack.py +130 -0
- maxframe/tensor/merge/tests/__init__.py +13 -0
- maxframe/tensor/merge/tests/test_merge.py +79 -0
- maxframe/tensor/merge/vstack.py +74 -0
- maxframe/tensor/misc/__init__.py +72 -0
- maxframe/tensor/misc/argwhere.py +72 -0
- maxframe/tensor/misc/array_split.py +46 -0
- maxframe/tensor/misc/astype.py +121 -0
- maxframe/tensor/misc/atleast_1d.py +72 -0
- maxframe/tensor/misc/atleast_2d.py +70 -0
- maxframe/tensor/misc/atleast_3d.py +85 -0
- maxframe/tensor/misc/broadcast_arrays.py +57 -0
- maxframe/tensor/misc/broadcast_to.py +89 -0
- maxframe/tensor/misc/copy.py +64 -0
- maxframe/tensor/misc/copyto.py +130 -0
- maxframe/tensor/misc/delete.py +104 -0
- maxframe/tensor/misc/diff.py +115 -0
- maxframe/tensor/misc/dsplit.py +68 -0
- maxframe/tensor/misc/ediff1d.py +74 -0
- maxframe/tensor/misc/expand_dims.py +85 -0
- maxframe/tensor/misc/flatten.py +63 -0
- maxframe/tensor/misc/flip.py +90 -0
- maxframe/tensor/misc/fliplr.py +64 -0
- maxframe/tensor/misc/flipud.py +68 -0
- maxframe/tensor/misc/hsplit.py +85 -0
- maxframe/tensor/misc/in1d.py +94 -0
- maxframe/tensor/misc/insert.py +139 -0
- maxframe/tensor/misc/isin.py +130 -0
- maxframe/tensor/misc/moveaxis.py +83 -0
- maxframe/tensor/misc/ndim.py +53 -0
- maxframe/tensor/misc/ravel.py +90 -0
- maxframe/tensor/misc/repeat.py +129 -0
- maxframe/tensor/misc/result_type.py +88 -0
- maxframe/tensor/misc/roll.py +124 -0
- maxframe/tensor/misc/rollaxis.py +77 -0
- maxframe/tensor/misc/searchsorted.py +147 -0
- maxframe/tensor/misc/setdiff1d.py +58 -0
- maxframe/tensor/misc/shape.py +89 -0
- maxframe/tensor/misc/split.py +190 -0
- maxframe/tensor/misc/squeeze.py +117 -0
- maxframe/tensor/misc/swapaxes.py +113 -0
- maxframe/tensor/misc/tests/__init__.py +13 -0
- maxframe/tensor/misc/tests/test_misc.py +112 -0
- maxframe/tensor/misc/tile.py +109 -0
- maxframe/tensor/misc/transpose.py +133 -0
- maxframe/tensor/misc/trapezoid.py +123 -0
- maxframe/tensor/misc/unique.py +227 -0
- maxframe/tensor/misc/vsplit.py +74 -0
- maxframe/tensor/misc/where.py +129 -0
- maxframe/tensor/operators.py +83 -0
- maxframe/tensor/random/__init__.py +166 -0
- maxframe/tensor/random/beta.py +87 -0
- maxframe/tensor/random/binomial.py +135 -0
- maxframe/tensor/random/bytes.py +37 -0
- maxframe/tensor/random/chisquare.py +108 -0
- maxframe/tensor/random/choice.py +187 -0
- maxframe/tensor/random/core.py +249 -0
- maxframe/tensor/random/dirichlet.py +121 -0
- maxframe/tensor/random/exponential.py +92 -0
- maxframe/tensor/random/f.py +133 -0
- maxframe/tensor/random/gamma.py +126 -0
- maxframe/tensor/random/geometric.py +91 -0
- maxframe/tensor/random/gumbel.py +165 -0
- maxframe/tensor/random/hypergeometric.py +146 -0
- maxframe/tensor/random/laplace.py +131 -0
- maxframe/tensor/random/logistic.py +127 -0
- maxframe/tensor/random/lognormal.py +157 -0
- maxframe/tensor/random/logseries.py +120 -0
- maxframe/tensor/random/multinomial.py +131 -0
- maxframe/tensor/random/multivariate_normal.py +190 -0
- maxframe/tensor/random/negative_binomial.py +123 -0
- maxframe/tensor/random/noncentral_chisquare.py +130 -0
- maxframe/tensor/random/noncentral_f.py +124 -0
- maxframe/tensor/random/normal.py +141 -0
- maxframe/tensor/random/pareto.py +138 -0
- maxframe/tensor/random/permutation.py +107 -0
- maxframe/tensor/random/poisson.py +109 -0
- maxframe/tensor/random/power.py +140 -0
- maxframe/tensor/random/rand.py +80 -0
- maxframe/tensor/random/randint.py +119 -0
- maxframe/tensor/random/randn.py +94 -0
- maxframe/tensor/random/random_integers.py +121 -0
- maxframe/tensor/random/random_sample.py +84 -0
- maxframe/tensor/random/rayleigh.py +108 -0
- maxframe/tensor/random/shuffle.py +61 -0
- maxframe/tensor/random/standard_cauchy.py +103 -0
- maxframe/tensor/random/standard_exponential.py +70 -0
- maxframe/tensor/random/standard_gamma.py +118 -0
- maxframe/tensor/random/standard_normal.py +72 -0
- maxframe/tensor/random/standard_t.py +133 -0
- maxframe/tensor/random/tests/__init__.py +13 -0
- maxframe/tensor/random/tests/test_random.py +165 -0
- maxframe/tensor/random/triangular.py +117 -0
- maxframe/tensor/random/uniform.py +129 -0
- maxframe/tensor/random/vonmises.py +129 -0
- maxframe/tensor/random/wald.py +112 -0
- maxframe/tensor/random/weibull.py +138 -0
- maxframe/tensor/random/zipf.py +120 -0
- maxframe/tensor/rechunk/__init__.py +26 -0
- maxframe/tensor/rechunk/rechunk.py +43 -0
- maxframe/tensor/reduction/__init__.py +64 -0
- maxframe/tensor/reduction/all.py +101 -0
- maxframe/tensor/reduction/allclose.py +86 -0
- maxframe/tensor/reduction/any.py +103 -0
- maxframe/tensor/reduction/argmax.py +101 -0
- maxframe/tensor/reduction/argmin.py +101 -0
- maxframe/tensor/reduction/array_equal.py +63 -0
- maxframe/tensor/reduction/core.py +166 -0
- maxframe/tensor/reduction/count_nonzero.py +80 -0
- maxframe/tensor/reduction/cumprod.py +95 -0
- maxframe/tensor/reduction/cumsum.py +99 -0
- maxframe/tensor/reduction/max.py +118 -0
- maxframe/tensor/reduction/mean.py +122 -0
- maxframe/tensor/reduction/min.py +118 -0
- maxframe/tensor/reduction/nanargmax.py +80 -0
- maxframe/tensor/reduction/nanargmin.py +74 -0
- maxframe/tensor/reduction/nancumprod.py +89 -0
- maxframe/tensor/reduction/nancumsum.py +92 -0
- maxframe/tensor/reduction/nanmax.py +109 -0
- maxframe/tensor/reduction/nanmean.py +105 -0
- maxframe/tensor/reduction/nanmin.py +109 -0
- maxframe/tensor/reduction/nanprod.py +92 -0
- maxframe/tensor/reduction/nanstd.py +124 -0
- maxframe/tensor/reduction/nansum.py +113 -0
- maxframe/tensor/reduction/nanvar.py +149 -0
- maxframe/tensor/reduction/prod.py +128 -0
- maxframe/tensor/reduction/std.py +132 -0
- maxframe/tensor/reduction/sum.py +123 -0
- maxframe/tensor/reduction/tests/__init__.py +13 -0
- maxframe/tensor/reduction/tests/test_reduction.py +189 -0
- maxframe/tensor/reduction/var.py +176 -0
- maxframe/tensor/reshape/__init__.py +15 -0
- maxframe/tensor/reshape/reshape.py +192 -0
- maxframe/tensor/reshape/tests/__init__.py +13 -0
- maxframe/tensor/reshape/tests/test_reshape.py +35 -0
- maxframe/tensor/sort/__init__.py +18 -0
- maxframe/tensor/sort/argpartition.py +98 -0
- maxframe/tensor/sort/argsort.py +150 -0
- maxframe/tensor/sort/partition.py +228 -0
- maxframe/tensor/sort/sort.py +295 -0
- maxframe/tensor/spatial/__init__.py +15 -0
- maxframe/tensor/spatial/distance/__init__.py +17 -0
- maxframe/tensor/spatial/distance/cdist.py +421 -0
- maxframe/tensor/spatial/distance/pdist.py +398 -0
- maxframe/tensor/spatial/distance/squareform.py +153 -0
- maxframe/tensor/special/__init__.py +175 -0
- maxframe/tensor/special/airy.py +55 -0
- maxframe/tensor/special/bessel.py +199 -0
- maxframe/tensor/special/core.py +99 -0
- maxframe/tensor/special/ellip_func_integrals.py +155 -0
- maxframe/tensor/special/ellip_harm.py +55 -0
- maxframe/tensor/special/err_fresnel.py +223 -0
- maxframe/tensor/special/gamma_funcs.py +303 -0
- maxframe/tensor/special/hypergeometric_funcs.py +69 -0
- maxframe/tensor/special/info_theory.py +189 -0
- maxframe/tensor/special/misc.py +163 -0
- maxframe/tensor/special/statistical.py +56 -0
- maxframe/tensor/statistics/__init__.py +24 -0
- maxframe/tensor/statistics/average.py +143 -0
- maxframe/tensor/statistics/bincount.py +133 -0
- maxframe/tensor/statistics/corrcoef.py +77 -0
- maxframe/tensor/statistics/cov.py +222 -0
- maxframe/tensor/statistics/digitize.py +126 -0
- maxframe/tensor/statistics/histogram.py +520 -0
- maxframe/tensor/statistics/median.py +85 -0
- maxframe/tensor/statistics/percentile.py +175 -0
- maxframe/tensor/statistics/ptp.py +89 -0
- maxframe/tensor/statistics/quantile.py +290 -0
- maxframe/tensor/ufunc/__init__.py +24 -0
- maxframe/tensor/ufunc/ufunc.py +198 -0
- maxframe/tensor/utils.py +719 -0
- maxframe/tests/__init__.py +13 -0
- maxframe/tests/test_protocol.py +178 -0
- maxframe/tests/test_udf.py +61 -0
- maxframe/tests/test_utils.py +627 -0
- maxframe/tests/utils.py +245 -0
- maxframe/typing_.py +42 -0
- maxframe/udf.py +435 -0
- maxframe/utils.py +1774 -0
- maxframe-2.4.0rc1.dist-info/METADATA +109 -0
- maxframe-2.4.0rc1.dist-info/RECORD +1122 -0
- maxframe-2.4.0rc1.dist-info/WHEEL +5 -0
- maxframe-2.4.0rc1.dist-info/top_level.txt +3 -0
- maxframe_client/__init__.py +16 -0
- maxframe_client/clients/__init__.py +13 -0
- maxframe_client/clients/framedriver.py +137 -0
- maxframe_client/conftest.py +15 -0
- maxframe_client/fetcher.py +411 -0
- maxframe_client/session/__init__.py +22 -0
- maxframe_client/session/consts.py +39 -0
- maxframe_client/session/graph.py +125 -0
- maxframe_client/session/odps.py +813 -0
- maxframe_client/session/task.py +329 -0
- maxframe_client/session/tests/__init__.py +13 -0
- maxframe_client/session/tests/test_task.py +115 -0
- maxframe_client/tests/__init__.py +13 -0
- maxframe_client/tests/test_fetcher.py +215 -0
- maxframe_client/tests/test_session.py +409 -0
|
@@ -0,0 +1,161 @@
|
|
|
1
|
+
# Copyright 1999-2025 Alibaba Group Holding Ltd.
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
|
|
15
|
+
from typing import Any, Tuple, Union
|
|
16
|
+
|
|
17
|
+
import numpy as np
|
|
18
|
+
import pandas as pd
|
|
19
|
+
import pyarrow as pa
|
|
20
|
+
|
|
21
|
+
from ...core import OutputType
|
|
22
|
+
from ...lib.version import parse as parse_version
|
|
23
|
+
from ...protocol import DataFrameTableMeta
|
|
24
|
+
from ...tensor.core import TENSOR_TYPE
|
|
25
|
+
from ...typing_ import ArrowTableType, PandasObjectTypes
|
|
26
|
+
from .schema import (
|
|
27
|
+
arrow_table_to_pandas_dataframe,
|
|
28
|
+
build_dataframe_table_meta,
|
|
29
|
+
pandas_dataframe_to_arrow_table,
|
|
30
|
+
)
|
|
31
|
+
|
|
32
|
+
|
|
33
|
+
def _rebuild_dataframe(
|
|
34
|
+
df: pd.DataFrame, table_meta: DataFrameTableMeta
|
|
35
|
+
) -> Union[pd.DataFrame, pd.Series]:
|
|
36
|
+
indexed = df.set_index(list(table_meta.table_index_column_names))
|
|
37
|
+
indexed.index.set_names(table_meta.pd_index_level_names, inplace=True)
|
|
38
|
+
if len(table_meta.pd_column_level_names) == 1:
|
|
39
|
+
cols = pd.Index(
|
|
40
|
+
table_meta.pd_column_names, name=table_meta.pd_column_level_names[0]
|
|
41
|
+
)
|
|
42
|
+
else:
|
|
43
|
+
cols = pd.MultiIndex.from_tuples(table_meta.pd_column_names).set_names(
|
|
44
|
+
table_meta.pd_column_level_names
|
|
45
|
+
)
|
|
46
|
+
indexed.columns = cols
|
|
47
|
+
if table_meta.type == OutputType.series:
|
|
48
|
+
return indexed.iloc[:, 0]
|
|
49
|
+
return indexed
|
|
50
|
+
|
|
51
|
+
|
|
52
|
+
def _rebuild_index(df: pd.DataFrame, table_meta: DataFrameTableMeta) -> pd.Index:
|
|
53
|
+
if df.shape[1] > 1:
|
|
54
|
+
idx = pd.MultiIndex.from_frame(df)
|
|
55
|
+
idx.names = table_meta.pd_index_level_names
|
|
56
|
+
else:
|
|
57
|
+
# make sure even if None names are updated properly
|
|
58
|
+
idx = pd.Index(df.iloc[:, 0])
|
|
59
|
+
idx.name = table_meta.pd_index_level_names[0]
|
|
60
|
+
return idx
|
|
61
|
+
|
|
62
|
+
|
|
63
|
+
def arrow_to_pandas(
|
|
64
|
+
arrow_table: ArrowTableType, table_meta: DataFrameTableMeta
|
|
65
|
+
) -> PandasObjectTypes:
|
|
66
|
+
df = arrow_table_to_pandas_dataframe(arrow_table, table_meta)
|
|
67
|
+
if table_meta.type in (OutputType.dataframe, OutputType.series):
|
|
68
|
+
return _rebuild_dataframe(df, table_meta)
|
|
69
|
+
elif table_meta.type == OutputType.index:
|
|
70
|
+
return _rebuild_index(df, table_meta)
|
|
71
|
+
elif table_meta.type == OutputType.tensor:
|
|
72
|
+
return _rebuild_index(df, table_meta).to_numpy()
|
|
73
|
+
elif table_meta.type == OutputType.scalar:
|
|
74
|
+
return _rebuild_index(df, table_meta)[0]
|
|
75
|
+
else: # this could never happen # pragma: no cover
|
|
76
|
+
raise ValueError(f"Does not support meta type {table_meta.type!r}")
|
|
77
|
+
|
|
78
|
+
|
|
79
|
+
def pandas_to_arrow(
|
|
80
|
+
df: Any, nthreads=1, ignore_index=False, ms_cols=None
|
|
81
|
+
) -> Tuple[ArrowTableType, DataFrameTableMeta]:
|
|
82
|
+
table_meta = build_dataframe_table_meta(df, ignore_index)
|
|
83
|
+
df = df.copy() if callable(getattr(df, "copy", None)) else df
|
|
84
|
+
table_datetime_cols = None
|
|
85
|
+
if table_meta.type in (OutputType.dataframe, OutputType.series):
|
|
86
|
+
if table_meta.type == OutputType.series:
|
|
87
|
+
df = df.to_frame("_data" if df.name is None else df.name)
|
|
88
|
+
if ms_cols:
|
|
89
|
+
table_datetime_cols = {"_data"}
|
|
90
|
+
elif ms_cols:
|
|
91
|
+
ms_col_set = set(ms_cols)
|
|
92
|
+
table_datetime_cols = set()
|
|
93
|
+
for pd_col, table_col in zip(
|
|
94
|
+
table_meta.pd_column_dtypes.keys(), table_meta.table_column_names
|
|
95
|
+
):
|
|
96
|
+
if pd_col in ms_col_set:
|
|
97
|
+
table_datetime_cols.add(table_col)
|
|
98
|
+
df.columns = pd.Index(table_meta.table_column_names)
|
|
99
|
+
if not ignore_index:
|
|
100
|
+
df = df.rename_axis(table_meta.table_index_column_names).reset_index()
|
|
101
|
+
elif ignore_index and table_meta.type != OutputType.index:
|
|
102
|
+
df = pd.DataFrame([], columns=[])
|
|
103
|
+
elif table_meta.type == OutputType.index:
|
|
104
|
+
names = [f"_idx_{idx}" for idx in range(len(df.names))]
|
|
105
|
+
table_datetime_cols = set()
|
|
106
|
+
if ms_cols:
|
|
107
|
+
if isinstance(df, pd.MultiIndex):
|
|
108
|
+
table_datetime_cols = {f"_idx_{idx}" for idx in ms_cols}
|
|
109
|
+
else:
|
|
110
|
+
table_datetime_cols = {"_idx_0"}
|
|
111
|
+
df = df.to_frame(name=names[0] if len(names) == 1 else names)
|
|
112
|
+
elif table_meta.type in (OutputType.scalar, OutputType.tensor):
|
|
113
|
+
names = ["_idx_0"]
|
|
114
|
+
if isinstance(df, (TENSOR_TYPE, np.ndarray)):
|
|
115
|
+
df = pd.DataFrame([], columns=names).astype({names[0]: df.dtype})
|
|
116
|
+
else:
|
|
117
|
+
df = pd.DataFrame([[df]], columns=names)
|
|
118
|
+
else: # this could never happen # pragma: no cover
|
|
119
|
+
raise ValueError(f"Does not support meta type {table_meta.type!r}")
|
|
120
|
+
|
|
121
|
+
try:
|
|
122
|
+
pa_table = pandas_dataframe_to_arrow_table(df, nthreads=nthreads)
|
|
123
|
+
except pa.ArrowTypeError as ex: # pragma: no cover
|
|
124
|
+
late_np_version = parse_version(np.__version__) >= parse_version("1.20")
|
|
125
|
+
early_pa_version = parse_version(pa.__version__) <= parse_version("4.0")
|
|
126
|
+
if (
|
|
127
|
+
late_np_version
|
|
128
|
+
and early_pa_version
|
|
129
|
+
and "Did not pass numpy.dtype object" in str(ex)
|
|
130
|
+
):
|
|
131
|
+
raise TypeError(
|
|
132
|
+
"Potential dependency conflict. Try update to pyarrow>4.0 "
|
|
133
|
+
"or downgrade to numpy<1.20. Details can be seen at "
|
|
134
|
+
"https://github.com/numpy/numpy/issues/17913. "
|
|
135
|
+
f"Raw error message: {ex!r}"
|
|
136
|
+
).with_traceback(ex.__traceback__) from None
|
|
137
|
+
else:
|
|
138
|
+
raise
|
|
139
|
+
|
|
140
|
+
if table_datetime_cols:
|
|
141
|
+
col_names = pa_table.schema.names
|
|
142
|
+
col_datas = []
|
|
143
|
+
for idx, col_name in enumerate(pa_table.schema.names):
|
|
144
|
+
if col_name not in table_datetime_cols:
|
|
145
|
+
col_datas.append(pa_table.column(idx))
|
|
146
|
+
continue
|
|
147
|
+
col_data = pa_table.column(idx).cast(pa.timestamp("ms"))
|
|
148
|
+
col_datas.append(col_data)
|
|
149
|
+
pa_table = pa.Table.from_arrays(col_datas, names=col_names)
|
|
150
|
+
|
|
151
|
+
new_names, new_dtypes = [], []
|
|
152
|
+
for table_col, (pd_col, pd_dtype) in zip(
|
|
153
|
+
table_meta.table_column_names, table_meta.pd_column_dtypes.items()
|
|
154
|
+
):
|
|
155
|
+
new_names.append(pd_col)
|
|
156
|
+
if table_col not in table_datetime_cols:
|
|
157
|
+
new_dtypes.append(pd_dtype)
|
|
158
|
+
else:
|
|
159
|
+
new_dtypes.append(np.dtype("datetime64[ms]"))
|
|
160
|
+
table_meta.pd_column_dtypes = pd.Series(new_dtypes, index=new_names)
|
|
161
|
+
return pa_table, table_meta
|
|
@@ -0,0 +1,533 @@
|
|
|
1
|
+
# Copyright 1999-2025 Alibaba Group Holding Ltd.
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
|
|
15
|
+
import string
|
|
16
|
+
from collections import defaultdict
|
|
17
|
+
from typing import Any, Dict, Tuple
|
|
18
|
+
|
|
19
|
+
import numpy as np
|
|
20
|
+
import pandas as pd
|
|
21
|
+
import pyarrow as pa
|
|
22
|
+
from odps import types as odps_types
|
|
23
|
+
from pandas.api import types as pd_types
|
|
24
|
+
|
|
25
|
+
from ...config import options
|
|
26
|
+
from ...core import TILEABLE_TYPE, OutputType
|
|
27
|
+
from ...dataframe.core import DATAFRAME_TYPE, INDEX_TYPE, SERIES_TYPE
|
|
28
|
+
from ...lib.dtypes_extension import ArrowBlobType, ArrowDtype
|
|
29
|
+
from ...protocol import DataFrameTableMeta
|
|
30
|
+
from ...tensor.core import TENSOR_TYPE
|
|
31
|
+
from ...utils import build_temp_table_name
|
|
32
|
+
|
|
33
|
+
_TEMP_TABLE_PREFIX = "tmp_mf_"
|
|
34
|
+
DEFAULT_SINGLE_INDEX_NAME = "_idx_0"
|
|
35
|
+
|
|
36
|
+
_arrow_to_odps_types = {
|
|
37
|
+
pa.string(): odps_types.string,
|
|
38
|
+
pa.binary(): odps_types.binary,
|
|
39
|
+
pa.int8(): odps_types.tinyint,
|
|
40
|
+
pa.int16(): odps_types.smallint,
|
|
41
|
+
pa.int32(): odps_types.int_,
|
|
42
|
+
pa.int64(): odps_types.bigint,
|
|
43
|
+
pa.bool_(): odps_types.boolean,
|
|
44
|
+
pa.float32(): odps_types.float_,
|
|
45
|
+
pa.float64(): odps_types.double,
|
|
46
|
+
pa.date32(): odps_types.date,
|
|
47
|
+
pa.timestamp("ms"): odps_types.datetime,
|
|
48
|
+
pa.timestamp("us"): odps_types.timestamp,
|
|
49
|
+
pa.timestamp("ns"): odps_types.timestamp,
|
|
50
|
+
}
|
|
51
|
+
|
|
52
|
+
_odps_type_to_arrow = {
|
|
53
|
+
odps_types.string: pa.string(),
|
|
54
|
+
odps_types.binary: pa.binary(),
|
|
55
|
+
odps_types.tinyint: pa.int8(),
|
|
56
|
+
odps_types.smallint: pa.int16(),
|
|
57
|
+
odps_types.int_: pa.int32(),
|
|
58
|
+
odps_types.bigint: pa.int64(),
|
|
59
|
+
odps_types.boolean: pa.bool_(),
|
|
60
|
+
odps_types.float_: pa.float32(),
|
|
61
|
+
odps_types.double: pa.float64(),
|
|
62
|
+
odps_types.date: pa.date32(),
|
|
63
|
+
odps_types.datetime: pa.timestamp("ms"),
|
|
64
|
+
odps_types.json: pa.string(),
|
|
65
|
+
odps_types.timestamp: pa.timestamp("ns"),
|
|
66
|
+
odps_types.timestamp_ntz: pa.timestamp("ns"),
|
|
67
|
+
}
|
|
68
|
+
|
|
69
|
+
_int_to_pd_dtype = {
|
|
70
|
+
np.dtype("int8"): pd.Int8Dtype(),
|
|
71
|
+
np.dtype("int16"): pd.Int16Dtype(),
|
|
72
|
+
np.dtype("int32"): pd.Int32Dtype(),
|
|
73
|
+
np.dtype("int64"): pd.Int64Dtype(),
|
|
74
|
+
np.dtype("uint8"): pd.UInt8Dtype(),
|
|
75
|
+
np.dtype("uint16"): pd.UInt16Dtype(),
|
|
76
|
+
np.dtype("uint32"): pd.UInt32Dtype(),
|
|
77
|
+
np.dtype("uint64"): pd.UInt64Dtype(),
|
|
78
|
+
}
|
|
79
|
+
|
|
80
|
+
if hasattr(odps_types, "blob"):
|
|
81
|
+
_arrow_to_odps_types[ArrowBlobType()] = odps_types.blob
|
|
82
|
+
_odps_type_to_arrow[odps_types.blob] = ArrowBlobType()
|
|
83
|
+
|
|
84
|
+
_based_for_pandas_pa_types = (pa.ListType, pa.MapType, pa.StructType)
|
|
85
|
+
|
|
86
|
+
|
|
87
|
+
def cast_df_with_possible_nans(
|
|
88
|
+
df_obj: pd.DataFrame, dtypes: pd.Series, try_cast_back: bool = True
|
|
89
|
+
) -> pd.DataFrame:
|
|
90
|
+
# align types to cast with existing dataframe
|
|
91
|
+
new_dtypes = df_obj.dtypes.copy()
|
|
92
|
+
new_dtypes.update(dtypes)
|
|
93
|
+
res = df_obj.astype(new_dtypes.replace(_int_to_pd_dtype))
|
|
94
|
+
if not try_cast_back:
|
|
95
|
+
return res
|
|
96
|
+
|
|
97
|
+
to_cast = {}
|
|
98
|
+
for idx, col_dtype in enumerate(new_dtypes):
|
|
99
|
+
if col_dtype in _int_to_pd_dtype and not res.iloc[:, idx].isna().any():
|
|
100
|
+
to_cast[idx] = col_dtype
|
|
101
|
+
if to_cast:
|
|
102
|
+
# in case there is duplications, only cast on integer indexes
|
|
103
|
+
res.columns, cols = range(len(res.columns)), res.columns
|
|
104
|
+
res = res.astype(to_cast)
|
|
105
|
+
res.columns = cols
|
|
106
|
+
return res
|
|
107
|
+
|
|
108
|
+
|
|
109
|
+
def is_based_for_pandas_dtype(arrow_type: pa.DataType) -> bool:
|
|
110
|
+
"""
|
|
111
|
+
Check whether the arrow type is based for one pandas data type.
|
|
112
|
+
If true, we should make sure the environment support ArrowDtype.
|
|
113
|
+
"""
|
|
114
|
+
if not isinstance(arrow_type, _based_for_pandas_pa_types):
|
|
115
|
+
return False
|
|
116
|
+
|
|
117
|
+
if ArrowDtype is None:
|
|
118
|
+
raise ImportError("ArrowDtype is not supported in current environment")
|
|
119
|
+
return True
|
|
120
|
+
|
|
121
|
+
|
|
122
|
+
def pandas_types_to_arrow_schema(df_obj: pd.DataFrame) -> pa.Schema:
|
|
123
|
+
"""
|
|
124
|
+
This one is only called when a pandas DataFrame is written to ODPS. So we can check
|
|
125
|
+
whether the ArrowDtype is supported.
|
|
126
|
+
"""
|
|
127
|
+
schema = pa.Schema.from_pandas(df_obj, preserve_index=False)
|
|
128
|
+
names, types = schema.names, schema.types
|
|
129
|
+
for idx, col_dtype in enumerate(df_obj.dtypes.tolist()):
|
|
130
|
+
if ArrowDtype is not None and isinstance(col_dtype, ArrowDtype):
|
|
131
|
+
types[idx] = col_dtype.pyarrow_dtype
|
|
132
|
+
elif isinstance(col_dtype, pd.StringDtype):
|
|
133
|
+
# as ODPS does not support large_string, we use string instead
|
|
134
|
+
types[idx] = pa.string()
|
|
135
|
+
return pa.schema([pa.field(name, typ) for name, typ in zip(names, types)])
|
|
136
|
+
|
|
137
|
+
|
|
138
|
+
def arrow_type_to_odps_type(
|
|
139
|
+
arrow_type: pa.DataType, col_name: str, unknown_as_string: bool = False
|
|
140
|
+
) -> odps_types.DataType:
|
|
141
|
+
if arrow_type in _arrow_to_odps_types:
|
|
142
|
+
return _arrow_to_odps_types[arrow_type]
|
|
143
|
+
elif isinstance(arrow_type, pa.ListType):
|
|
144
|
+
return odps_types.Array(
|
|
145
|
+
arrow_type_to_odps_type(arrow_type.value_type, col_name, unknown_as_string)
|
|
146
|
+
)
|
|
147
|
+
elif isinstance(arrow_type, pa.MapType):
|
|
148
|
+
return odps_types.Map(
|
|
149
|
+
arrow_type_to_odps_type(arrow_type.key_type, col_name, unknown_as_string),
|
|
150
|
+
arrow_type_to_odps_type(arrow_type.item_type, col_name, unknown_as_string),
|
|
151
|
+
)
|
|
152
|
+
elif isinstance(arrow_type, pa.StructType):
|
|
153
|
+
type_dict = {}
|
|
154
|
+
for idx in range(arrow_type.num_fields):
|
|
155
|
+
field = arrow_type[idx]
|
|
156
|
+
type_dict[field.name] = arrow_type_to_odps_type(
|
|
157
|
+
field.type, col_name, unknown_as_string
|
|
158
|
+
)
|
|
159
|
+
return odps_types.Struct(type_dict)
|
|
160
|
+
elif isinstance(arrow_type, pa.Decimal128Type):
|
|
161
|
+
return odps_types.Decimal(arrow_type.precision, arrow_type.scale)
|
|
162
|
+
|
|
163
|
+
if unknown_as_string:
|
|
164
|
+
return odps_types.string
|
|
165
|
+
else:
|
|
166
|
+
raise TypeError(
|
|
167
|
+
"Unknown type {}, column name is {},"
|
|
168
|
+
"specify `unknown_as_string=True` "
|
|
169
|
+
"or `as_type` to set column dtype".format(arrow_type, col_name)
|
|
170
|
+
)
|
|
171
|
+
|
|
172
|
+
|
|
173
|
+
def arrow_schema_to_odps_schema(
|
|
174
|
+
arrow_schema: pa.Schema, unknown_as_string: bool = False
|
|
175
|
+
) -> odps_types.OdpsSchema:
|
|
176
|
+
odps_cols = []
|
|
177
|
+
for col_name, col_type in zip(arrow_schema.names, arrow_schema.types):
|
|
178
|
+
col_odps_type = arrow_type_to_odps_type(
|
|
179
|
+
col_type, col_name, unknown_as_string=unknown_as_string
|
|
180
|
+
)
|
|
181
|
+
odps_cols.append(odps_types.Column(col_name, col_odps_type))
|
|
182
|
+
|
|
183
|
+
return odps_types.OdpsSchema(odps_cols)
|
|
184
|
+
|
|
185
|
+
|
|
186
|
+
def odps_type_to_arrow_type(
|
|
187
|
+
odps_type: odps_types.DataType, col_name: str
|
|
188
|
+
) -> pa.DataType:
|
|
189
|
+
if odps_type in _odps_type_to_arrow:
|
|
190
|
+
col_type = _odps_type_to_arrow[odps_type]
|
|
191
|
+
else:
|
|
192
|
+
if isinstance(odps_type, odps_types.Array):
|
|
193
|
+
col_type = pa.list_(odps_type_to_arrow_type(odps_type.value_type, col_name))
|
|
194
|
+
elif isinstance(odps_type, odps_types.Map):
|
|
195
|
+
col_type = pa.map_(
|
|
196
|
+
odps_type_to_arrow_type(odps_type.key_type, col_name),
|
|
197
|
+
odps_type_to_arrow_type(odps_type.value_type, col_name),
|
|
198
|
+
)
|
|
199
|
+
elif isinstance(odps_type, odps_types.Struct):
|
|
200
|
+
fields = [
|
|
201
|
+
(k, odps_type_to_arrow_type(v, col_name))
|
|
202
|
+
for k, v in odps_type.field_types.items()
|
|
203
|
+
]
|
|
204
|
+
col_type = pa.struct(fields)
|
|
205
|
+
elif isinstance(odps_type, odps_types.Decimal):
|
|
206
|
+
if odps_type.name == "decimal":
|
|
207
|
+
# legacy decimal data without precision or scale
|
|
208
|
+
# precision data from internal compat mode
|
|
209
|
+
col_type = pa.decimal128(38, 18)
|
|
210
|
+
else:
|
|
211
|
+
col_type = pa.decimal128(
|
|
212
|
+
odps_type.precision or odps_types.Decimal._max_precision,
|
|
213
|
+
odps_type.scale or odps_types.Decimal._max_scale,
|
|
214
|
+
)
|
|
215
|
+
elif isinstance(odps_type, (odps_types.Varchar, odps_types.Char)):
|
|
216
|
+
col_type = pa.string()
|
|
217
|
+
else:
|
|
218
|
+
raise TypeError(
|
|
219
|
+
"Unsupported type {}, column name is {}".format(odps_type, col_name)
|
|
220
|
+
)
|
|
221
|
+
return col_type
|
|
222
|
+
|
|
223
|
+
|
|
224
|
+
def odps_schema_to_arrow_schema(
|
|
225
|
+
odps_schema: odps_types.OdpsSchema, with_partitions: bool = False
|
|
226
|
+
) -> pa.Schema:
|
|
227
|
+
arrow_schema = []
|
|
228
|
+
cols = odps_schema.columns if with_partitions else odps_schema.simple_columns
|
|
229
|
+
for col in cols:
|
|
230
|
+
col_name = col.name
|
|
231
|
+
col_type = odps_type_to_arrow_type(col.type, col_name)
|
|
232
|
+
arrow_schema.append(pa.field(col_name, col_type))
|
|
233
|
+
|
|
234
|
+
return pa.schema(arrow_schema)
|
|
235
|
+
|
|
236
|
+
|
|
237
|
+
def odps_schema_to_pandas_dtypes(
|
|
238
|
+
odps_schema: odps_types.OdpsSchema, with_partitions: bool = False
|
|
239
|
+
) -> pd.Series:
|
|
240
|
+
arrow_schema = odps_schema_to_arrow_schema(
|
|
241
|
+
odps_schema, with_partitions=with_partitions
|
|
242
|
+
)
|
|
243
|
+
return arrow_table_to_pandas_dataframe(arrow_schema.empty_table()).dtypes
|
|
244
|
+
|
|
245
|
+
|
|
246
|
+
def arrow_table_to_pandas_dataframe(
|
|
247
|
+
table: pa.Table, meta: DataFrameTableMeta = None
|
|
248
|
+
) -> pd.DataFrame:
|
|
249
|
+
use_arrow_backend = options.dataframe.dtype_backend == "pyarrow"
|
|
250
|
+
df = table.to_pandas(
|
|
251
|
+
types_mapper=lambda x: (
|
|
252
|
+
ArrowDtype(x) if is_based_for_pandas_dtype(x) or use_arrow_backend else None
|
|
253
|
+
),
|
|
254
|
+
ignore_metadata=True,
|
|
255
|
+
)
|
|
256
|
+
if not meta:
|
|
257
|
+
return df
|
|
258
|
+
|
|
259
|
+
# If meta is passed, we should convert the dtypes to match the ones in the meta
|
|
260
|
+
converted_column_dtypes = dict()
|
|
261
|
+
for source_dtype, target_col, target_dtype in zip(
|
|
262
|
+
df.dtypes.values,
|
|
263
|
+
df.columns,
|
|
264
|
+
list(meta.pd_index_dtypes.values) + list(meta.pd_column_dtypes.values),
|
|
265
|
+
):
|
|
266
|
+
if source_dtype != target_dtype:
|
|
267
|
+
# Converting tz-aware dtype to tz-native dtype is a special case.
|
|
268
|
+
# In numpy1.19, we can't use numpy.dtype.DateTime64Dtype
|
|
269
|
+
if (
|
|
270
|
+
isinstance(source_dtype, pd.DatetimeTZDtype)
|
|
271
|
+
and isinstance(target_dtype, np.dtype)
|
|
272
|
+
and target_dtype.name.startswith("datetime64")
|
|
273
|
+
):
|
|
274
|
+
df[target_col] = df[target_col].dt.tz_localize(None)
|
|
275
|
+
else:
|
|
276
|
+
converted_column_dtypes[target_col] = target_dtype
|
|
277
|
+
|
|
278
|
+
if converted_column_dtypes:
|
|
279
|
+
df = cast_df_with_possible_nans(df, converted_column_dtypes)
|
|
280
|
+
|
|
281
|
+
return df
|
|
282
|
+
|
|
283
|
+
|
|
284
|
+
def pandas_dataframe_to_arrow_table(df: pd.DataFrame, nthreads=1) -> pa.Table:
|
|
285
|
+
schema = pandas_types_to_arrow_schema(df)
|
|
286
|
+
return pa.Table.from_pandas(
|
|
287
|
+
df, schema=schema, nthreads=nthreads, preserve_index=False
|
|
288
|
+
)
|
|
289
|
+
|
|
290
|
+
|
|
291
|
+
def pandas_dtypes_to_arrow_schema(dtypes, unknown_as_string: bool = False) -> pa.Schema:
|
|
292
|
+
if unknown_as_string:
|
|
293
|
+
dt_list = [dt if dt != np.dtype("O") else pd.StringDtype() for dt in dtypes]
|
|
294
|
+
dtypes = pd.Series(dt_list, index=dtypes.index)
|
|
295
|
+
schema = pandas_types_to_arrow_schema(
|
|
296
|
+
pd.DataFrame([], columns=dtypes.index).astype(dtypes)
|
|
297
|
+
)
|
|
298
|
+
return schema
|
|
299
|
+
|
|
300
|
+
|
|
301
|
+
def pandas_dtype_to_arrow_type(dtype, unknown_as_string: bool = False) -> pa.DataType:
|
|
302
|
+
if unknown_as_string and dtype == np.dtype("O"):
|
|
303
|
+
dtype = pd.StringDtype()
|
|
304
|
+
schema = pandas_types_to_arrow_schema(
|
|
305
|
+
pd.DataFrame([], columns=["a"]).astype({"a": dtype})
|
|
306
|
+
)
|
|
307
|
+
return schema.types[0]
|
|
308
|
+
|
|
309
|
+
|
|
310
|
+
def is_scalar_object(df_obj: Any) -> bool:
|
|
311
|
+
return (
|
|
312
|
+
isinstance(df_obj, TENSOR_TYPE) and df_obj.shape == ()
|
|
313
|
+
) or pd_types.is_scalar(df_obj)
|
|
314
|
+
|
|
315
|
+
|
|
316
|
+
def is_tensor_object(df_obj: Any) -> bool:
|
|
317
|
+
return (
|
|
318
|
+
isinstance(df_obj, TENSOR_TYPE) or isinstance(df_obj, np.ndarray)
|
|
319
|
+
) and df_obj.ndim <= 1
|
|
320
|
+
|
|
321
|
+
|
|
322
|
+
def _scalar_as_index(df_obj: Any) -> pd.Index:
|
|
323
|
+
if isinstance(df_obj, TILEABLE_TYPE):
|
|
324
|
+
return pd.Index([], dtype=df_obj.dtype)
|
|
325
|
+
else:
|
|
326
|
+
return pd.Index([df_obj])[:0]
|
|
327
|
+
|
|
328
|
+
|
|
329
|
+
def pandas_to_odps_schema(
|
|
330
|
+
df_obj: Any,
|
|
331
|
+
unknown_as_string: bool = False,
|
|
332
|
+
ignore_index=False,
|
|
333
|
+
) -> Tuple[odps_types.OdpsSchema, DataFrameTableMeta]:
|
|
334
|
+
from ... import dataframe as md
|
|
335
|
+
from .arrow import pandas_to_arrow
|
|
336
|
+
|
|
337
|
+
if is_scalar_object(df_obj) or is_tensor_object(df_obj):
|
|
338
|
+
empty_index = None
|
|
339
|
+
elif hasattr(df_obj, "index_value"):
|
|
340
|
+
empty_index = df_obj.index_value.to_pandas()[:0]
|
|
341
|
+
elif not isinstance(df_obj, pd.Index):
|
|
342
|
+
empty_index = df_obj.index[:0]
|
|
343
|
+
else:
|
|
344
|
+
empty_index = df_obj[:0]
|
|
345
|
+
|
|
346
|
+
if hasattr(df_obj, "columns_value"):
|
|
347
|
+
empty_columns = df_obj.dtypes.index
|
|
348
|
+
elif hasattr(df_obj, "columns"):
|
|
349
|
+
empty_columns = df_obj.columns
|
|
350
|
+
else:
|
|
351
|
+
empty_columns = None
|
|
352
|
+
|
|
353
|
+
ms_cols = None
|
|
354
|
+
if isinstance(df_obj, (md.DataFrame, pd.DataFrame)):
|
|
355
|
+
empty_df_obj = pd.DataFrame(
|
|
356
|
+
[], columns=empty_columns, index=empty_index
|
|
357
|
+
).astype(df_obj.dtypes)
|
|
358
|
+
ms_cols = [
|
|
359
|
+
col for col, dt in df_obj.dtypes.items() if dt == np.dtype("datetime64[ms]")
|
|
360
|
+
]
|
|
361
|
+
elif isinstance(df_obj, (md.Series, pd.Series)):
|
|
362
|
+
empty_df_obj = pd.Series([], name=df_obj.name, index=empty_index).astype(
|
|
363
|
+
df_obj.dtype
|
|
364
|
+
)
|
|
365
|
+
ms_cols = df_obj.dtype == np.dtype("datetime64[ms]")
|
|
366
|
+
elif isinstance(df_obj, (md.Index, pd.Index)):
|
|
367
|
+
empty_df_obj = empty_index
|
|
368
|
+
if isinstance(empty_index, pd.MultiIndex):
|
|
369
|
+
ms_cols = [
|
|
370
|
+
idx
|
|
371
|
+
for idx, dt in enumerate(empty_index.dtypes.values)
|
|
372
|
+
if dt == np.dtype("datetime64[ms]")
|
|
373
|
+
]
|
|
374
|
+
else:
|
|
375
|
+
ms_cols = df_obj.dtype == np.dtype("datetime64[ms]")
|
|
376
|
+
else:
|
|
377
|
+
empty_df_obj = df_obj
|
|
378
|
+
|
|
379
|
+
arrow_data, table_meta = pandas_to_arrow(
|
|
380
|
+
empty_df_obj, ignore_index=ignore_index, ms_cols=ms_cols
|
|
381
|
+
)
|
|
382
|
+
return (
|
|
383
|
+
arrow_schema_to_odps_schema(
|
|
384
|
+
arrow_data.schema, unknown_as_string=unknown_as_string
|
|
385
|
+
),
|
|
386
|
+
table_meta,
|
|
387
|
+
)
|
|
388
|
+
|
|
389
|
+
|
|
390
|
+
def build_table_column_name(
|
|
391
|
+
col_idx: int, pd_col_name: Any, records: Dict[str, str]
|
|
392
|
+
) -> str:
|
|
393
|
+
"""
|
|
394
|
+
Convert column name to MaxCompute acceptable names
|
|
395
|
+
|
|
396
|
+
Parameters
|
|
397
|
+
----------
|
|
398
|
+
col_idx:
|
|
399
|
+
index of the column
|
|
400
|
+
pd_col_name:
|
|
401
|
+
column name in pandas
|
|
402
|
+
records:
|
|
403
|
+
record for existing columns
|
|
404
|
+
|
|
405
|
+
Returns
|
|
406
|
+
-------
|
|
407
|
+
converted column name
|
|
408
|
+
"""
|
|
409
|
+
|
|
410
|
+
def _is_col_name_legal(name: str):
|
|
411
|
+
if len(name) < 1 or len(name) > 128:
|
|
412
|
+
return False
|
|
413
|
+
if name[0] not in string.ascii_letters and name[0] != "_":
|
|
414
|
+
return False
|
|
415
|
+
for ch in name:
|
|
416
|
+
if ch not in string.digits and ch not in string.ascii_letters and ch != "_":
|
|
417
|
+
return False
|
|
418
|
+
return True
|
|
419
|
+
|
|
420
|
+
try:
|
|
421
|
+
return records[pd_col_name]
|
|
422
|
+
except KeyError:
|
|
423
|
+
pass
|
|
424
|
+
|
|
425
|
+
if isinstance(pd_col_name, str):
|
|
426
|
+
col_name = pd_col_name
|
|
427
|
+
elif isinstance(pd_col_name, tuple):
|
|
428
|
+
col_name = "_".join(str(x) for x in pd_col_name)
|
|
429
|
+
else:
|
|
430
|
+
col_name = str(pd_col_name)
|
|
431
|
+
|
|
432
|
+
col_name = col_name.lower()
|
|
433
|
+
if not _is_col_name_legal(col_name):
|
|
434
|
+
col_name = f"_column_{col_idx}"
|
|
435
|
+
|
|
436
|
+
records[pd_col_name] = col_name
|
|
437
|
+
return col_name
|
|
438
|
+
|
|
439
|
+
|
|
440
|
+
def build_dataframe_table_meta(
|
|
441
|
+
df_obj: Any, ignore_index: bool = False, session_id: str = None
|
|
442
|
+
) -> DataFrameTableMeta:
|
|
443
|
+
col_to_count = defaultdict(lambda: 0)
|
|
444
|
+
col_to_idx = defaultdict(lambda: 0)
|
|
445
|
+
pd_col_to_col_name = dict()
|
|
446
|
+
if isinstance(df_obj, (DATAFRAME_TYPE, pd.DataFrame)):
|
|
447
|
+
obj_type = OutputType.dataframe
|
|
448
|
+
elif isinstance(df_obj, (SERIES_TYPE, pd.Series)):
|
|
449
|
+
obj_type = OutputType.series
|
|
450
|
+
elif isinstance(df_obj, (INDEX_TYPE, pd.Index)):
|
|
451
|
+
obj_type = OutputType.index
|
|
452
|
+
elif is_tensor_object(df_obj) and df_obj.ndim == 1:
|
|
453
|
+
obj_type = OutputType.tensor
|
|
454
|
+
elif is_scalar_object(df_obj):
|
|
455
|
+
obj_type = OutputType.scalar
|
|
456
|
+
else: # pragma: no cover
|
|
457
|
+
raise TypeError(f"Cannot accept type {type(df_obj)}")
|
|
458
|
+
|
|
459
|
+
if obj_type in (OutputType.scalar, OutputType.tensor):
|
|
460
|
+
pd_dtypes = pd.Series([])
|
|
461
|
+
column_index_names = []
|
|
462
|
+
index_obj = _scalar_as_index(df_obj)
|
|
463
|
+
elif obj_type == OutputType.index:
|
|
464
|
+
pd_dtypes = pd.Series([])
|
|
465
|
+
column_index_names = []
|
|
466
|
+
index_obj = df_obj
|
|
467
|
+
elif obj_type == OutputType.series:
|
|
468
|
+
pd_dtypes = pd.Series([df_obj.dtype], index=[df_obj.name])
|
|
469
|
+
column_index_names = [None]
|
|
470
|
+
index_obj = df_obj.index
|
|
471
|
+
else:
|
|
472
|
+
pd_dtypes = df_obj.dtypes
|
|
473
|
+
column_index_names = list(pd_dtypes.index.names)
|
|
474
|
+
index_obj = df_obj.index
|
|
475
|
+
|
|
476
|
+
if isinstance(df_obj, TILEABLE_TYPE):
|
|
477
|
+
if not session_id:
|
|
478
|
+
table_name = _TEMP_TABLE_PREFIX + str(df_obj.key)
|
|
479
|
+
else:
|
|
480
|
+
table_name = build_temp_table_name(session_id, df_obj.key)
|
|
481
|
+
else:
|
|
482
|
+
table_name = None
|
|
483
|
+
|
|
484
|
+
sql_columns = [None] * len(pd_dtypes)
|
|
485
|
+
pd_col_names = pd_dtypes.index
|
|
486
|
+
if obj_type == OutputType.series and df_obj.name is None:
|
|
487
|
+
# use special table column name for series
|
|
488
|
+
pd_col_names = ["_data"]
|
|
489
|
+
for idx, col in enumerate(pd_col_names):
|
|
490
|
+
sql_columns[idx] = col_name = build_table_column_name(
|
|
491
|
+
idx, col, pd_col_to_col_name
|
|
492
|
+
)
|
|
493
|
+
col_to_count[col_name] += 1
|
|
494
|
+
|
|
495
|
+
final_sql_columns = []
|
|
496
|
+
for col in sql_columns:
|
|
497
|
+
if col_to_count[col] > 1:
|
|
498
|
+
col_name = f"{col}_{col_to_idx[col]}"
|
|
499
|
+
col_to_idx[col] += 1
|
|
500
|
+
while col_name in col_to_count:
|
|
501
|
+
col_name = f"{col}_{col_to_idx[col]}"
|
|
502
|
+
col_to_idx[col] += 1
|
|
503
|
+
final_sql_columns.append(col_name)
|
|
504
|
+
else:
|
|
505
|
+
final_sql_columns.append(col)
|
|
506
|
+
|
|
507
|
+
if hasattr(index_obj, "index_value"):
|
|
508
|
+
pd_index_val = index_obj.index_value.to_pandas()
|
|
509
|
+
else:
|
|
510
|
+
pd_index_val = index_obj
|
|
511
|
+
|
|
512
|
+
level_dtypes = [
|
|
513
|
+
pd_index_val.get_level_values(level).dtype
|
|
514
|
+
for level in range(pd_index_val.nlevels)
|
|
515
|
+
]
|
|
516
|
+
index_dtypes = pd.Series(level_dtypes, index=pd_index_val.names)
|
|
517
|
+
|
|
518
|
+
if ignore_index and obj_type != OutputType.index:
|
|
519
|
+
table_index_column_names = []
|
|
520
|
+
pd_index_dtypes = pd.Series([], index=[])
|
|
521
|
+
else:
|
|
522
|
+
table_index_column_names = [f"_idx_{i}" for i in range(len(index_obj.names))]
|
|
523
|
+
pd_index_dtypes = index_dtypes
|
|
524
|
+
|
|
525
|
+
return DataFrameTableMeta(
|
|
526
|
+
table_name=table_name,
|
|
527
|
+
type=obj_type,
|
|
528
|
+
table_column_names=final_sql_columns,
|
|
529
|
+
table_index_column_names=table_index_column_names,
|
|
530
|
+
pd_column_dtypes=pd_dtypes,
|
|
531
|
+
pd_column_level_names=column_index_names,
|
|
532
|
+
pd_index_dtypes=pd_index_dtypes,
|
|
533
|
+
)
|