polars-runtime-compat 1.34.0b3__cp39-abi3-manylinux_2_24_aarch64.whl → 1.34.0b4__cp39-abi3-manylinux_2_24_aarch64.whl

polars/functions/__init__.py

@@ -1,193 +0,0 @@
-from polars.functions.aggregation import (
-    all,
-    all_horizontal,
-    any,
-    any_horizontal,
-    cum_sum,
-    cum_sum_horizontal,
-    max,
-    max_horizontal,
-    mean_horizontal,
-    min,
-    min_horizontal,
-    sum,
-    sum_horizontal,
-)
-from polars.functions.as_datatype import (
-    concat_arr,
-    concat_list,
-    concat_str,
-    duration,
-    format,
-    struct,
-)
-from polars.functions.as_datatype import date_ as date
-from polars.functions.as_datatype import datetime_ as datetime
-from polars.functions.as_datatype import time_ as time
-from polars.functions.business import business_day_count
-from polars.functions.col import col
-from polars.functions.datatype import dtype_of, self_dtype, struct_with_fields
-from polars.functions.eager import align_frames, concat
-from polars.functions.escape_regex import escape_regex
-from polars.functions.lazy import (
-    _row_encode,
-    approx_n_unique,
-    arctan2,
-    arctan2d,
-    arg_sort_by,
-    arg_where,
-    coalesce,
-    collect_all,
-    collect_all_async,
-    corr,
-    count,
-    cov,
-    cum_count,
-    cum_fold,
-    cum_reduce,
-    element,
-    exclude,
-    explain_all,
-    field,
-    first,
-    fold,
-    from_epoch,
-    groups,
-    head,
-    implode,
-    last,
-    map_batches,
-    map_groups,
-    mean,
-    median,
-    n_unique,
-    nth,
-    quantile,
-    reduce,
-    rolling_corr,
-    rolling_cov,
-    row_index,
-    select,
-    sql_expr,
-    std,
-    tail,
-    var,
-)
-from polars.functions.len import len
-from polars.functions.lit import lit
-from polars.functions.random import set_random_seed
-from polars.functions.range import (
-    arange,
-    date_range,
-    date_ranges,
-    datetime_range,
-    datetime_ranges,
-    int_range,
-    int_ranges,
-    linear_space,
-    linear_spaces,
-    time_range,
-    time_ranges,
-)
-from polars.functions.repeat import ones, repeat, zeros
-from polars.functions.whenthen import when
-
-__all__ = [
-    # polars.functions.aggregation
-    "all",
-    "any",
-    "cum_sum",
-    "max",
-    "min",
-    "sum",
-    "all_horizontal",
-    "any_horizontal",
-    "cum_sum_horizontal",
-    "max_horizontal",
-    "min_horizontal",
-    "sum_horizontal",
-    # polars.functions.datatype
-    "dtype_of",
-    "self_dtype",
-    "struct_with_fields",
-    # polars.functions.eager
-    "align_frames",
-    "approx_n_unique",
-    "arg_where",
-    "concat",
-    "date_range",
-    "date_ranges",
-    "datetime_range",
-    "datetime_ranges",
-    "element",
-    "ones",
-    "repeat",
-    "time_range",
-    "time_ranges",
-    "zeros",
-    # polars.functions.lazy
-    "_row_encode",
-    "arange",
-    "arctan2",
-    "arctan2d",
-    "arg_sort_by",
-    "business_day_count",
-    "coalesce",
-    "col",
-    "collect_all",
-    "collect_all_async",
-    "concat_arr",
-    "concat_list",
-    "concat_str",
-    "corr",
-    "count",
-    "cov",
-    "cum_count",
-    "cum_fold",
-    "cum_reduce",
-    "date",  # named date_, see import above
-    "datetime",  # named datetime_, see import above
-    "duration",
-    "exclude",
-    "explain_all",
-    "field",
-    "first",
-    "fold",
-    "format",
-    "from_epoch",
-    "groups",
-    "head",
-    "implode",
-    "int_range",
-    "int_ranges",
-    "last",
-    "linear_space",
-    "linear_spaces",
-    "lit",
-    "map_batches",
-    "map_groups",
-    "mean",
-    "mean_horizontal",
-    "median",
-    "n_unique",
-    "nth",
-    "quantile",
-    "reduce",
-    "rolling_corr",
-    "rolling_cov",
-    "row_index",
-    "select",
-    "set_random_seed",
-    "std",
-    "struct",
-    "tail",
-    "time",
-    "var",
-    # polars.functions.len
-    "len",
-    # polars.functions.whenthen
-    "when",
-    "sql_expr",
-    # polars.functions.escape_regex
-    "escape_regex",
-]

polars/functions/aggregation/__init__.py

@@ -1,33 +0,0 @@
-from polars.functions.aggregation.horizontal import (
-    all_horizontal,
-    any_horizontal,
-    cum_sum_horizontal,
-    max_horizontal,
-    mean_horizontal,
-    min_horizontal,
-    sum_horizontal,
-)
-from polars.functions.aggregation.vertical import (
-    all,
-    any,
-    cum_sum,
-    max,
-    min,
-    sum,
-)
-
-__all__ = [
-    "all",
-    "all_horizontal",
-    "any",
-    "any_horizontal",
-    "cum_sum",
-    "cum_sum_horizontal",
-    "max",
-    "max_horizontal",
-    "mean_horizontal",
-    "min",
-    "min_horizontal",
-    "sum",
-    "sum_horizontal",
-]

polars/functions/aggregation/horizontal.py

@@ -1,298 +0,0 @@
-from __future__ import annotations
-
-import contextlib
-from typing import TYPE_CHECKING
-
-import polars.functions as F
-from polars._utils.parse import parse_into_list_of_expressions
-from polars._utils.wrap import wrap_expr
-
-with contextlib.suppress(ImportError):  # Module not available when building docs
-    import polars._plr as plr
-
-if TYPE_CHECKING:
-    from collections.abc import Iterable
-
-    from polars import Expr
-    from polars._typing import IntoExpr
-
-
-def all_horizontal(*exprs: IntoExpr | Iterable[IntoExpr]) -> Expr:
-    """
-    Compute the bitwise AND horizontally across columns.
-
-    Parameters
-    ----------
-    *exprs
-        Column(s) to use in the aggregation. Accepts expression input. Strings are
-        parsed as column names, other non-expression inputs are parsed as literals.
-
-    Notes
-    -----
-    `Kleene logic`_ is used to deal with nulls: if the column contains any null values
-    and no `False` values, the output is null.
-
-    .. _Kleene logic: https://en.wikipedia.org/wiki/Three-valued_logic
-
-    Examples
-    --------
-    >>> df = pl.DataFrame(
-    ...     {
-    ...         "a": [False, False, True, True, False, None],
-    ...         "b": [False, True, True, None, None, None],
-    ...         "c": ["u", "v", "w", "x", "y", "z"],
-    ...     }
-    ... )
-    >>> df.with_columns(all=pl.all_horizontal("a", "b"))
-    shape: (6, 4)
-    ┌───────┬───────┬─────┬───────┐
-    │ a     ┆ b     ┆ c   ┆ all   │
-    │ ---   ┆ ---   ┆ --- ┆ ---   │
-    │ bool  ┆ bool  ┆ str ┆ bool  │
-    ╞═══════╪═══════╪═════╪═══════╡
-    │ false ┆ false ┆ u   ┆ false │
-    │ false ┆ true  ┆ v   ┆ false │
-    │ true  ┆ true  ┆ w   ┆ true  │
-    │ true  ┆ null  ┆ x   ┆ null  │
-    │ false ┆ null  ┆ y   ┆ false │
-    │ null  ┆ null  ┆ z   ┆ null  │
-    └───────┴───────┴─────┴───────┘
-    """
-    pyexprs = parse_into_list_of_expressions(*exprs)
-    return wrap_expr(plr.all_horizontal(pyexprs))
-
-
-def any_horizontal(*exprs: IntoExpr | Iterable[IntoExpr]) -> Expr:
-    """
-    Compute the bitwise OR horizontally across columns.
-
-    Parameters
-    ----------
-    *exprs
-        Column(s) to use in the aggregation. Accepts expression input. Strings are
-        parsed as column names, other non-expression inputs are parsed as literals.
-
-    Notes
-    -----
-    `Kleene logic`_ is used to deal with nulls: if the column contains any null values
-    and no `True` values, the output is null.
-
-    .. _Kleene logic: https://en.wikipedia.org/wiki/Three-valued_logic
-
-    Examples
-    --------
-    >>> df = pl.DataFrame(
-    ...     {
-    ...         "a": [False, False, True, True, False, None],
-    ...         "b": [False, True, True, None, None, None],
-    ...         "c": ["u", "v", "w", "x", "y", "z"],
-    ...     }
-    ... )
-    >>> df.with_columns(any=pl.any_horizontal("a", "b"))
-    shape: (6, 4)
-    ┌───────┬───────┬─────┬───────┐
-    │ a     ┆ b     ┆ c   ┆ any   │
-    │ ---   ┆ ---   ┆ --- ┆ ---   │
-    │ bool  ┆ bool  ┆ str ┆ bool  │
-    ╞═══════╪═══════╪═════╪═══════╡
-    │ false ┆ false ┆ u   ┆ false │
-    │ false ┆ true  ┆ v   ┆ true  │
-    │ true  ┆ true  ┆ w   ┆ true  │
-    │ true  ┆ null  ┆ x   ┆ true  │
-    │ false ┆ null  ┆ y   ┆ null  │
-    │ null  ┆ null  ┆ z   ┆ null  │
-    └───────┴───────┴─────┴───────┘
-    """
-    pyexprs = parse_into_list_of_expressions(*exprs)
-    return wrap_expr(plr.any_horizontal(pyexprs))
-
-
-def max_horizontal(*exprs: IntoExpr | Iterable[IntoExpr]) -> Expr:
-    """
-    Get the maximum value horizontally across columns.
-
-    Parameters
-    ----------
-    *exprs
-        Column(s) to use in the aggregation. Accepts expression input. Strings are
-        parsed as column names, other non-expression inputs are parsed as literals.
-
-    Examples
-    --------
-    >>> df = pl.DataFrame(
-    ...     {
-    ...         "a": [1, 8, 3],
-    ...         "b": [4, 5, None],
-    ...         "c": ["x", "y", "z"],
-    ...     }
-    ... )
-    >>> df.with_columns(max=pl.max_horizontal("a", "b"))
-    shape: (3, 4)
-    ┌─────┬──────┬─────┬─────┐
-    │ a   ┆ b    ┆ c   ┆ max │
-    │ --- ┆ ---  ┆ --- ┆ --- │
-    │ i64 ┆ i64  ┆ str ┆ i64 │
-    ╞═════╪══════╪═════╪═════╡
-    │ 1   ┆ 4    ┆ x   ┆ 4   │
-    │ 8   ┆ 5    ┆ y   ┆ 8   │
-    │ 3   ┆ null ┆ z   ┆ 3   │
-    └─────┴──────┴─────┴─────┘
-    """
-    pyexprs = parse_into_list_of_expressions(*exprs)
-    return wrap_expr(plr.max_horizontal(pyexprs))
-
-
-def min_horizontal(*exprs: IntoExpr | Iterable[IntoExpr]) -> Expr:
-    """
-    Get the minimum value horizontally across columns.
-
-    Parameters
-    ----------
-    *exprs
-        Column(s) to use in the aggregation. Accepts expression input. Strings are
-        parsed as column names, other non-expression inputs are parsed as literals.
-
-    Examples
-    --------
-    >>> df = pl.DataFrame(
-    ...     {
-    ...         "a": [1, 8, 3],
-    ...         "b": [4, 5, None],
-    ...         "c": ["x", "y", "z"],
-    ...     }
-    ... )
-    >>> df.with_columns(min=pl.min_horizontal("a", "b"))
-    shape: (3, 4)
-    ┌─────┬──────┬─────┬─────┐
-    │ a   ┆ b    ┆ c   ┆ min │
-    │ --- ┆ ---  ┆ --- ┆ --- │
-    │ i64 ┆ i64  ┆ str ┆ i64 │
-    ╞═════╪══════╪═════╪═════╡
-    │ 1   ┆ 4    ┆ x   ┆ 1   │
-    │ 8   ┆ 5    ┆ y   ┆ 5   │
-    │ 3   ┆ null ┆ z   ┆ 3   │
-    └─────┴──────┴─────┴─────┘
-    """
-    pyexprs = parse_into_list_of_expressions(*exprs)
-    return wrap_expr(plr.min_horizontal(pyexprs))
-
-
-def sum_horizontal(
-    *exprs: IntoExpr | Iterable[IntoExpr], ignore_nulls: bool = True
-) -> Expr:
-    """
-    Sum all values horizontally across columns.
-
-    Parameters
-    ----------
-    *exprs
-        Column(s) to use in the aggregation. Accepts expression input. Strings are
-        parsed as column names, other non-expression inputs are parsed as literals.
-    ignore_nulls
-        Ignore null values (default).
-        If set to `False`, any null value in the input will lead to a null output.
-
-    Examples
-    --------
-    >>> df = pl.DataFrame(
-    ...     {
-    ...         "a": [1, 8, 3],
-    ...         "b": [4, 5, None],
-    ...         "c": ["x", "y", "z"],
-    ...     }
-    ... )
-    >>> df.with_columns(sum=pl.sum_horizontal("a", "b"))
-    shape: (3, 4)
-    ┌─────┬──────┬─────┬─────┐
-    │ a   ┆ b    ┆ c   ┆ sum │
-    │ --- ┆ ---  ┆ --- ┆ --- │
-    │ i64 ┆ i64  ┆ str ┆ i64 │
-    ╞═════╪══════╪═════╪═════╡
-    │ 1   ┆ 4    ┆ x   ┆ 5   │
-    │ 8   ┆ 5    ┆ y   ┆ 13  │
-    │ 3   ┆ null ┆ z   ┆ 3   │
-    └─────┴──────┴─────┴─────┘
-    """
-    pyexprs = parse_into_list_of_expressions(*exprs)
-    return wrap_expr(plr.sum_horizontal(pyexprs, ignore_nulls))
-
-
-def mean_horizontal(
-    *exprs: IntoExpr | Iterable[IntoExpr], ignore_nulls: bool = True
-) -> Expr:
-    """
-    Compute the mean of all values horizontally across columns.
-
-    Parameters
-    ----------
-    *exprs
-        Column(s) to use in the aggregation. Accepts expression input. Strings are
-        parsed as column names, other non-expression inputs are parsed as literals.
-    ignore_nulls
-        Ignore null values (default).
-        If set to `False`, any null value in the input will lead to a null output.
-
-    Examples
-    --------
-    >>> df = pl.DataFrame(
-    ...     {
-    ...         "a": [1, 8, 3],
-    ...         "b": [4, 5, None],
-    ...         "c": ["x", "y", "z"],
-    ...     }
-    ... )
-    >>> df.with_columns(mean=pl.mean_horizontal("a", "b"))
-    shape: (3, 4)
-    ┌─────┬──────┬─────┬──────┐
-    │ a   ┆ b    ┆ c   ┆ mean │
-    │ --- ┆ ---  ┆ --- ┆ ---  │
-    │ i64 ┆ i64  ┆ str ┆ f64  │
-    ╞═════╪══════╪═════╪══════╡
-    │ 1   ┆ 4    ┆ x   ┆ 2.5  │
-    │ 8   ┆ 5    ┆ y   ┆ 6.5  │
-    │ 3   ┆ null ┆ z   ┆ 3.0  │
-    └─────┴──────┴─────┴──────┘
-    """
-    pyexprs = parse_into_list_of_expressions(*exprs)
-    return wrap_expr(plr.mean_horizontal(pyexprs, ignore_nulls))
-
-
-def cum_sum_horizontal(*exprs: IntoExpr | Iterable[IntoExpr]) -> Expr:
-    """
-    Cumulatively sum all values horizontally across columns.
-
-    Parameters
-    ----------
-    *exprs
-        Column(s) to use in the aggregation. Accepts expression input. Strings are
-        parsed as column names, other non-expression inputs are parsed as literals.
-
-    Examples
-    --------
-    >>> df = pl.DataFrame(
-    ...     {
-    ...         "a": [1, 8, 3],
-    ...         "b": [4, 5, None],
-    ...         "c": ["x", "y", "z"],
-    ...     }
-    ... )
-    >>> df.with_columns(pl.cum_sum_horizontal("a", "b"))
-    shape: (3, 4)
-    ┌─────┬──────┬─────┬───────────┐
-    │ a   ┆ b    ┆ c   ┆ cum_sum   │
-    │ --- ┆ ---  ┆ --- ┆ ---       │
-    │ i64 ┆ i64  ┆ str ┆ struct[2] │
-    ╞═════╪══════╪═════╪═══════════╡
-    │ 1   ┆ 4    ┆ x   ┆ {1,5}     │
-    │ 8   ┆ 5    ┆ y   ┆ {8,13}    │
-    │ 3   ┆ null ┆ z   ┆ {3,null}  │
-    └─────┴──────┴─────┴───────────┘
-    """
-    pyexprs = parse_into_list_of_expressions(*exprs)
-    exprs_wrapped = [wrap_expr(e) for e in pyexprs]
-
-    return F.cum_fold(
-        F.lit(0).cast(F.dtype_of(F.sum_horizontal(list(exprs)))),
-        lambda a, b: a + b,
-        exprs_wrapped,
-    ).alias("cum_sum")