it%~ddlmZddlZddlmZmZddlmZddl m Z ddl m Z ddl mZeje5ddlmZdddddlmZerdd lmZmZdd lmZmZmZmZmZed d d  dd Zed d ddZed d ddZe ddd  ddZed d d d ddZed d d ddZed d d ddZdddd ddZy#1swYxYw) ) annotationsN) TYPE_CHECKINGoverload) functions)parse_into_expression)unstable) wrap_expr)Literal)ExprSeries)ClosedIntervalIntoExprIntoExprColumnNumericLiteralTemporalLiteral.)closedeagercyNstartend num_samplesrrs e/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/functions/range/linear_space.py linear_spacers )rcyrrrs rrr&srcyrrrs rrr1srbothFct|}t|}t|}ttj||||}|r#t j |j S|S)u Create sequence of evenly-spaced points. Parameters ---------- start Lower bound of the range. end Upper bound of the range. num_samples Number of samples in the output sequence. closed : {'both', 'left', 'right', 'none'} Define which sides of the interval are closed (inclusive). eager Evaluate immediately and return a `Series`. If set to `False` (default), return an expression instead. .. warning:: This functionality is experimental. It may be changed at any point without it being considered a breaking change. Notes ----- `linear_space` works with numeric and temporal dtypes. When the `start` and `end` parameters are `Date` dtypes, the output sequence consists of equally-spaced `Datetime` elements with millisecond precision. Returns ------- Expr or Series Column of data type `:class:Time`. Examples -------- >>> pl.linear_space(start=0, end=1, num_samples=3, eager=True) shape: (3,) Series: 'literal' [f64] [ 0.0 0.5 1.0 ] >>> pl.linear_space(start=0, end=1, num_samples=3, closed="left", eager=True) shape: (3,) Series: 'literal' [f64] [ 0.0 0.333333 0.666667 ] >>> pl.linear_space(start=0, end=1, num_samples=3, closed="right", eager=True) shape: (3,) Series: 'literal' [f64] [ 0.333333 0.666667 1.0 ] >>> pl.linear_space(start=0, end=1, num_samples=3, closed="none", eager=True) shape: (3,) Series: 'literal' [f64] [ 0.25 0.5 0.75 ] >>> from datetime import time >>> pl.linear_space( ... start=time(hour=1), end=time(hour=12), num_samples=3, eager=True ... ) shape: (3,) Series: 'literal' [time] [ 01:00:00 06:30:00 12:00:00 ] `Date` endpoints generate a sequence of `Datetime` values: >>> from datetime import date >>> pl.linear_space( ... start=date(2025, 1, 1), ... end=date(2025, 2, 1), ... num_samples=3, ... closed="right", ... eager=True, ... ) shape: (3,) Series: 'literal' [datetime[μs]] [ 2025-01-11 08:00:00 2025-01-21 16:00:00 2025-02-01 00:00:00 ] When `eager=False` (default), an expression is produced. You can generate a sequence using the length of the dataframe: >>> df = pl.DataFrame({"a": [1, 2, 3, 4, 5]}) >>> df.with_columns(pl.linear_space(0, 1, pl.len()).alias("ls")) shape: (5, 2) ┌─────┬──────┐ │ a ┆ ls │ │ --- ┆ --- │ │ i64 ┆ f64 │ ╞═════╪══════╡ │ 1 ┆ 0.0 │ │ 2 ┆ 0.25 │ │ 3 ┆ 0.5 │ │ 4 ┆ 0.75 │ │ 5 ┆ 1.0 │ └─────┴──────┘ )rr plrrFselect to_series) rrrrr start_pyexpr end_pyexprnum_samples_pyexprresults rrr<sev)/L&s+J.{;  z3EvNF xx))++ Mr)ras_arrayrcyrrrrrrr*rs r linear_spacesr-s r)rr*cyrrr,s rr-r-srcyrrr,s rr-r-src t|}t|}t|}ttj|||||} |r#t j | j S| S)u{ Generate a sequence of evenly-spaced values for each row between `start` and `end`. The number of values in each sequence is determined by `num_samples`. Parameters ---------- start Lower bound of the range. end Upper bound of the range. num_samples Number of samples in the output sequence. closed : {'both', 'left', 'right', 'none'} Define which sides of the interval are closed (inclusive). as_array Return result as a fixed-length `Array`. `num_samples` must be a constant. eager Evaluate immediately and return a `Series`. If set to `False` (default), return an expression instead. .. warning:: This functionality is experimental. It may be changed at any point without it being considered a breaking change. Returns ------- Expr or Series Column of data type `List(dtype)`. See Also -------- linear_space : Generate a single sequence of linearly-spaced values. Examples -------- >>> df = pl.DataFrame({"start": [1, -1], "end": [3, 2], "num_samples": [4, 5]}) >>> df.with_columns(ls=pl.linear_spaces("start", "end", "num_samples")) shape: (2, 4) ┌───────┬─────┬─────────────┬────────────────────────┐ │ start ┆ end ┆ num_samples ┆ ls │ │ --- ┆ --- ┆ --- ┆ --- │ │ i64 ┆ i64 ┆ i64 ┆ list[f64] │ ╞═══════╪═════╪═════════════╪════════════════════════╡ │ 1 ┆ 3 ┆ 4 ┆ [1.0, 1.666667, … 3.0] │ │ -1 ┆ 2 ┆ 5 ┆ [-1.0, -0.25, … 2.0] │ └───────┴─────┴─────────────┴────────────────────────┘ >>> df.with_columns(ls=pl.linear_spaces("start", "end", 3, as_array=True)) shape: (2, 4) ┌───────┬─────┬─────────────┬──────────────────┐ │ start ┆ end ┆ num_samples ┆ ls │ │ --- ┆ --- ┆ --- ┆ --- │ │ i64 ┆ i64 ┆ i64 ┆ array[f64, 3] │ ╞═══════╪═════╪═════════════╪══════════════════╡ │ 1 ┆ 3 ┆ 4 ┆ [1.0, 2.0, 3.0] │ │ -1 ┆ 2 ┆ 5 ┆ [-1.0, 0.5, 2.0] │ └───────┴─────┴─────────────┴──────────────────┘ )rr r"r-r#r$r%) rrrrr*rr&r'r(r)s rr-r-sjF)/L&s+J.{;   *&8&( F  xx))++ Mr) r+NumericLiteral | TemporalLiteral | IntoExprrr1rint | IntoExprrr rLiteral[False]returnr ) rr1rr1rr2rr r Literal[True]r4r ) rr1rr1rr2rr rboolr4 Expr | Series)r1NumericLiteral | TemporalLiteral | IntoExprColumnrr8rint | IntoExprColumnrr r*r6rr3r4r )rr8rr8rr9rr r*r6rr5r4r )rr8rr8rr9rr r*r6rr6r4r7) __future__r contextlibtypingrrpolarsrr#polars._utils.parserpolars._utils.unstablerpolars._utils.wrapr suppress ImportError polars._plr_plrr"r r r polars._typingr rrrrrr-rrrrFs$"*!5+(Z%# !  6 4        !  6 4        !  6 4        $ D 6D 4D D  D  DD DN ! < :&        !  < :&        !  < :&       $O <O :O&O  O  O OO{s D33D<