o ¹­§i#ã@s"ddlmZddlZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZddlmZe e¡ddlmZWdƒn1sHwYergdd lmZdd lmZmZmZdd lmZmZe d0d d d œd1dd„ƒZe d0d dœd2dd„ƒZe d0d dœd3d"d„ƒZ  #d4ed$d œd3d%d„Ze d0d d d œd1d&d'„ƒZe d0d dœd2d(d'„ƒZe d0d dœd3d)d'„ƒZ  #d4ed$d œd3d*d'„Ze d0d d d œd5d+d,„ƒZ e d0d dœd6d-d,„ƒZ e d0d dœd7d.d,„ƒZ  #d4ed$d œd7d/d,„Z dS)8é)Ú annotationsN)Ú TYPE_CHECKINGÚoverload)Ú functions)Úparse_into_expression)Ú wrap_exprÚwrap_s)ÚInt64)Úparse_into_datatype_expr)ÚLiteral)Ú DataTypeExprÚExprÚSeries)ÚIntoExprColumnÚPolarsIntegerType.©ÚdtypeÚeagerÚstartúint | IntoExprColumnÚendúint | IntoExprColumn | NoneÚstepÚintrú PolarsIntegerType | DataTypeExprrúLiteral[False]Úreturnr cCódS©N©©rrrrrrrúW/home/app/Keep/.python/lib/python3.10/site-packages/polars/functions/range/int_range.pyÚarangeór")rú Literal[True]rcCrrrr rrr!r"!r#Úboolú Expr | SeriescCrrrr rrr!r",r#éFcCst|||||dS)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_ranger rrr!r"7s0cCrrrr rrr!r(jr#r(cCrrrr rrr!r(ur#cCrrrr rrr!r(€r#c Cs‚|dur|}d}t|ƒ}t|tƒr#t|tƒr#|r#tt ||||j¡ƒSt|ƒ}t|ƒ}tt  ||||j¡ƒ}|r?t   |¡  ¡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) r Ú isinstancerrÚplrZeager_int_rangeÚ_pydatatype_exprrrr(ÚFÚselectÚ to_series) rrrrrÚ dtype_exprÚ start_pyexprÚ end_pyexprÚresultrrr!r(‹s KÿÿcCrrrr rrr!Ú int_rangesìr#r3cCrrrr rrr!r3÷r#cCrrrr rrr!r3r#c Cs\|dur|}d}t|ƒ}t|ƒ}t|ƒ}t|ƒ}tt ||||j¡ƒ} |r,t | ¡ ¡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) r rrr*r3r+r,r-r.) rrrrrr/r0r1Z step_pyexprr2rrr!r3 s> ÿÿ)...) rrrrrrrrrrrr ) rrrrrrrrrr$rr) rrrrrrrrrr%rr&)rNr') rrrrrrrrrrrr ) rrrrrrrrrr$rr) rrrrrrrrrr%rr&)!Ú __future__rÚ contextlibÚtypingrrZpolarsrr,Zpolars._utils.parserZpolars._utils.wraprrZpolars.datatypesr Zpolars.datatypes._parser ÚsuppressÚ ImportErrorZ polars._plrZ_plrr*r r r rZpolars._typingrrr"r(r3rrrr!ÚsÌ      ÿ ýú ýû ýû ýú3ýú ýû ýû ýúaýú ýû ýû ýú