i#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!dZ e dd d d"dZ e dd d d#dZ d edd  d#dZ y#1swY`xYw)$) annotationsN) TYPE_CHECKINGoverload) functions)parse_into_expression) wrap_exprwrap_s)Int64)parse_into_datatype_expr)Literal) DataTypeExprExprSeries)IntoExprColumnPolarsIntegerType.dtypeeagercyNstartendsteprrs b/var/www/pod-logistic/pod-ai/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_series) rrrrr dtype_expr start_pyexpr end_pyexprresults rr%r%sV {)%0J%*S#"65   sD*2M2M N  )/L&s+J  lJj6Q6QRF xx))++ Mrcyrrrs r int_rangesr7rrcyrrrs rr7r7r!rcyrrrs rr7r7r#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,r7r.r/r0r1) rrrrrr2r3r4 step_pyexprr5s rr7r7 s| {)%0J(/L&s+J'-K   *k:3N3N F  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>rrAr@r) rr<rr=rr<rr>rrBr@rC)! __future__r contextlibtypingrrpolarsrr/polars._utils.parserpolars._utils.wraprr polars.datatypesr polars.datatypes._parser suppress ImportError polars._plr_plrr,r r rrpolars._typingrrrr%r7rrrrRs"*!50"<Z%11@ "%'* /2   $  ,      "%'* /2   $  ,      "%'* /2   $  ,    #$'+0A /4 0A 0A $0A 0A , 0A  0A0Af "%'* /2   $  ,      "%'* /2   $  ,      "%'* /2   $  ,    #$'+^ /4 ^ ^ $^ ^ , ^  ^^B "%'*!$ /2   $  ,      "%'*!$ /2   $  ,      "%'*!$ /2   $  ,    #$'+!"O /4 O O $O O , O  OOCs F**F4