ˀh"nddlmZddlZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZddlmZej$e5ddlmZddderdd lmZdd lmZmZmZdd lmZmZe dd d d  ddZe dd d ddZe dd d ddZ d edd  ddZe dd d d  ddZe dd d ddZe dd d ddZ d edd  ddZe dd d d  d!dZe dd d d"dZe dd d d#dZ d edd  d#dZy#1swY`xYw)$) annotationsN) TYPE_CHECKINGoverload) functions)parse_into_expression) wrap_exprwrap_s)Int64)parse_into_datatype_expr)Literal) DataTypeExprExprSeries)IntoExprColumnPolarsIntegerType.dtypeeagercyNstartendsteprrs |/var/www/html/wine-match-dev/backend/winematch-backend/venv/lib/python3.12/site-packages/polars/functions/range/int_range.pyaranger )rcyrrrs rrr!rcyrrrs rrr,rFc"t|||||S)aS Generate a range of integers. Alias for :func:`int_range`. Parameters ---------- start Lower bound of the range (inclusive). end Upper bound of the range (exclusive). step Step size of the range. dtype Data type of the range. Defaults to `Int64`. eager Evaluate immediately and return a `Series`. If set to `False` (default), return an expression instead. Returns ------- Expr or Series Column of integer data type `dtype`. See Also -------- int_range : Generate a range of integers. int_ranges : Generate a range of integers for each row of the input columns. Examples -------- >>> pl.arange(0, 3, eager=True) shape: (3,) Series: 'literal' [i64] [ 0 1 2 ] r) int_rangers rrr7s` UCU% @@rcyrrrs rr%r%jrrcyrrrs rr%r%ur!rcyrrrs rr%r%r#rc||}d}t|}t|tr=t|tr-|r+tt j ||||j St|}t|}tt j||||j }|r#tj|jS|S)u Generate a range of integers. Parameters ---------- start Start of the range (inclusive). Defaults to 0. end End of the range (exclusive). If set to `None` (default), the value of `start` is used and `start` is set to `0`. step Step size of the range. dtype Data type of the range. eager Evaluate immediately and return a `Series`. If set to `False` (default), return an expression instead. Returns ------- Expr or Series Column of integer data type `dtype`. See Also -------- int_ranges : Generate a range of integers for each row of the input columns. Examples -------- >>> pl.int_range(0, 3, eager=True) shape: (3,) Series: 'literal' [i64] [ 0 1 2 ] `end` can be omitted for a shorter syntax. >>> pl.int_range(3, eager=True) shape: (3,) Series: 'literal' [i64] [ 0 1 2 ] Generate an index column by using `int_range` in conjunction with :func:`len`. >>> df = pl.DataFrame({"a": [1, 3, 5], "b": [2, 4, 6]}) >>> df.select( ... pl.int_range(pl.len(), dtype=pl.UInt32).alias("index"), ... pl.all(), ... ) shape: (3, 3) ┌───────┬─────┬─────┐ │ index ┆ a ┆ b │ │ --- ┆ --- ┆ --- │ │ u32 ┆ i64 ┆ i64 │ ╞═══════╪═════╪═════╡ │ 0 ┆ 1 ┆ 2 │ │ 1 ┆ 3 ┆ 4 │ │ 2 ┆ 5 ┆ 6 │ └───────┴─────┴─────┘ r) r isinstanceintr plreager_int_range_pydatatype_exprrrr%Fselect to_seriesrrrrr dtype_exprresults rr%r%sV {)%0J%*S#"65   sD*2M2M N   "% (E  $C s}}UCz7R7RS TF xx))++ Mrcyrrrs r int_rangesr6rrcyrrrs rr6r6r!rcyrrrs rr6r6r#rc ||}d}t|}t|}t|}t|}ttj||||j }|r#t j|jS|S)u Generate a range of integers for each row of the input columns. Parameters ---------- start Start of the range (inclusive). Defaults to 0. end End of the range (exclusive). If set to `None` (default), the value of `start` is used and `start` is set to `0`. step Step size of the range. dtype Integer data type of the ranges. Defaults to `Int64`. eager Evaluate immediately and return a `Series`. If set to `False` (default), return an expression instead. Returns ------- Expr or Series Column of data type `List(dtype)`. See Also -------- int_range : Generate a single range of integers. Examples -------- >>> df = pl.DataFrame({"start": [1, -1], "end": [3, 2]}) >>> df.with_columns(int_range=pl.int_ranges("start", "end")) shape: (2, 3) ┌───────┬─────┬────────────┐ │ start ┆ end ┆ int_range │ │ --- ┆ --- ┆ --- │ │ i64 ┆ i64 ┆ list[i64] │ ╞═══════╪═════╪════════════╡ │ 1 ┆ 3 ┆ [1, 2] │ │ -1 ┆ 2 ┆ [-1, 0, 1] │ └───────┴─────┴────────────┘ `end` can be omitted for a shorter syntax. >>> df.select("end", int_range=pl.int_ranges("end")) shape: (2, 2) ┌─────┬───────────┐ │ end ┆ int_range │ │ --- ┆ --- │ │ i64 ┆ list[i64] │ ╞═════╪═══════════╡ │ 3 ┆ [0, 1, 2] │ │ 2 ┆ [0, 1] │ └─────┴───────────┘ r) r rrr,r6r.r/r0r1r2s rr6r6 s{| {)%0J !% (E  $C  &D s~~eS$ 8S8ST UF xx))++ Mr)...) rint | IntoExprColumnrint | IntoExprColumn | Nonerr+r PolarsIntegerType | DataTypeExprrLiteral[False]returnr) rr:rr;rr+rr<r Literal[True]r>r) rr:rr;rr+rr<rboolr> Expr | Series)rN) rr:rr;rr:rr<rr=r>r) rr:rr;rr:rr<rr?r>r) rr:rr;rr:rr<rr@r>rA) __future__r contextlibtypingrrpolarsrr/polars._utils.parserpolars._utils.wraprr polars.datatypesr polars.datatypes._parser suppress ImportError polars.polarsr,r r rrpolars._typingrrrr%r6rrrrOs"*!50"<Z%  11@ "%'* /2   $  ,      "%'* /2   $  ,      "%'* /2   $  ,    #$'+0A /4 0A 0A $0A 0A , 0A  0A0Af "%'* /2   $  ,      "%'* /2   $  ,      "%'* /2   $  ,    #$'+\ /4 \ \ $\ \ , \  \\~ "%'*!$ /2   $  ,      "%'*!$ /2   $  ,      "%'*!$ /2   $  ,    #$'+!"K /4 K K $K K , K  KK  s F**F4