q-Ph ddlmZddlZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZeje5ddlmZdddn #1swxYwYerddlmZdd lmZmZdd lmZmZe d,d d d d-dZe d,d dd.dZe d,d dd/d Z d0ed"d d/d#Ze d,d d d d-d$Ze d,d dd.d%Ze d,d dd/d&Z d0ed"d d/d'Ze d,d d d d1d(Ze d,d dd2d)Ze d,d dd3d*Z d0ed"d d3d+ZdS)4) annotationsN) TYPE_CHECKINGoverload) functions)parse_into_expression) wrap_exprwrap_s)Int64)Literal)ExprSeries)IntoExprColumnPolarsIntegerType.dtypeeagerstartint | IntoExprColumnendint | IntoExprColumn | NonestepintrrrLiteral[False]returnr cdSNrrrrrs `/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/functions/range/int_range.pyaranger   3)r Literal[True]r cdSrrrs rr r Sr"bool Expr | SeriescdSrrrs rr r + Cr"Fc*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 rr r 6s` UCU% @ @ @@r"cdSrrrs rr,r,ir!r"cdSrrrs rr,r,tr%r"cdSrrrs rr,r,r)r"c||}d}t|tr;t|tr&|r$ttj||||St |}t |}t tj||||}|r&tj | S|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 │ └───────┴─────┴─────┘ Nr) isinstancerr plreager_int_rangerrr,Fselect to_seriesrrrrrresults rr,r,sV {%D*S#"6"6D5Dc)%dEBBCCC !% ( (E  $ $C s}UCu== > >F ,x))+++ Mr"cdSrrrs r int_rangesr:r!r"cdSrrrs rr:r:r%r"cdSrrrs rr:r:r)r"c||}d}t|}t|}t|}ttj||||}|r&t j|S|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] │ └─────┴───────────┘ Nr)rrr2r:r4r5r6r7s rr:r:s| { !% ( (E  $ $C  & &D s~eS$>> ? ?F ,x))+++ Mr")...) rrrrrrrrrrrr ) rrrrrrrrrr#rr ) rrrrrrrrrr&rr')rNr*) rrrrrrrrrrrr ) rrrrrrrrrr#rr ) rrrrrrrrrr&rr') __future__r contextlibtypingrrpolarsrr4polars._utils.parserpolars._utils.wraprr polars.datatypesr suppress ImportError polars.polarsr2r r r polars._typingrrr r,r:rr"rrIsC""""""********!!!!!!55555500000000""""""Z%%                 A########@@@@@@@@ "%'* #   "%'* #   "%'* #  #$'+0A % 0A0A0A0A0A0Af "%'* #   "%'* #   "%'* #  #$'+Y % YYYYYYx "%'*!$ #   "%'*!$ #   "%'*!$ #  #$'+!"J % JJJJJJJJsAAA