q-Ph#ddlmZddlZddlmZmZddlmZddl m Z ddl m Z ddl mZeje5ddlmZdddn #1swxYwYer ddlmZmZmZdd lmZdd lmZmZdd lmZmZe d)d d d d*dZe d)d dd+dZe d)d dd,d Z d-d"d#d d,d$Ze d)d d d d*d%Ze d)d dd+d&Ze d)d dd,d'Z d-d"d#d d,d(ZdS).) annotationsN) TYPE_CHECKINGoverload) functions)parse_into_expression) wrap_expr)parse_interval_argument)datedatetime timedelta)Literal)ExprSeries)ClosedIntervalIntoExprColumn.)closedeagerstart date | datetime | IntoExprColumnendintervalstr | timedeltarrrLiteral[False]returnrcdSNrrrrrs a/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/functions/range/date_range.py date_ranger   3)r Literal[True]rcdSrrrs rr r ! Sr"bool Series | ExprcdSrrrs rr r , Cr"1dbothFct|}t|}t|}ttj||||}|r&t j|S|S)u3 Generate a date range. Parameters ---------- start Lower bound of the date range. end Upper bound of the date range. interval Interval of the range periods, specified as a Python `timedelta` object or using the Polars duration string language (see "Notes" section below). Must consist of full days. closed : {'both', 'left', 'right', 'none'} Define which sides of the range are closed (inclusive). eager Evaluate immediately and return a `Series`. If set to `False` (default), return an expression instead. Returns ------- Expr or Series Column of data type :class:`Date`. See Also -------- date_ranges datetime_range Notes ----- `interval` is created according to the following string language: - 1d (1 calendar day) - 1w (1 calendar week) - 1mo (1 calendar month) - 1q (1 calendar quarter) - 1y (1 calendar year) Or combine them: "1w2d" # 1 week, 2 days By "calendar day", we mean the corresponding time on the next day (which may not be 24 hours, due to daylight savings). Similarly for "calendar week", "calendar month", "calendar quarter", and "calendar year". Examples -------- Using Polars duration string to specify the interval: >>> from datetime import date >>> pl.date_range(date(2022, 1, 1), date(2022, 3, 1), "1mo", eager=True).alias( ... "date" ... ) shape: (3,) Series: 'date' [date] [ 2022-01-01 2022-02-01 2022-03-01 ] Using `timedelta` object to specify the interval: >>> from datetime import timedelta >>> pl.date_range( ... date(1985, 1, 1), ... date(1985, 1, 10), ... timedelta(days=2), ... eager=True, ... ).alias("date") shape: (5,) Series: 'date' [date] [ 1985-01-01 1985-01-03 1985-01-05 1985-01-07 1985-01-09 ] Omit `eager=True` if you want to use `date_range` as an expression: >>> df = pl.DataFrame( ... { ... "date": [ ... date(2024, 1, 1), ... date(2024, 1, 2), ... date(2024, 1, 1), ... date(2024, 1, 3), ... ], ... "key": ["one", "one", "two", "two"], ... } ... ) >>> result = ( ... df.group_by("key") ... .agg(pl.date_range(pl.col("date").min(), pl.col("date").max())) ... .sort("key") ... ) >>> with pl.Config(fmt_str_lengths=50): ... print(result) shape: (2, 2) ┌─────┬──────────────────────────────────────┐ │ key ┆ date │ │ --- ┆ --- │ │ str ┆ list[date] │ ╞═════╪══════════════════════════════════════╡ │ one ┆ [2024-01-01, 2024-01-02] │ │ two ┆ [2024-01-01, 2024-01-02, 2024-01-03] │ └─────┴──────────────────────────────────────┘ )r rrplrr Fselect to_seriesrrrrr start_pyexpr end_pyexprresults rr r 7srn'x00H(//L&s++J s~lJ&QQ R RF ,x))+++ Mr"cdSrrrs r date_rangesr6r!r"cdSrrrs rr6r6r%r"cdSrrrs rr6r6r)r"ct|}t|}t|}ttj||||}|r&t j|S|S)u[ Create a column of date ranges. Parameters ---------- start Lower bound of the date range. end Upper bound of the date range. interval Interval of the range periods, specified as a Python `timedelta` object or using the Polars duration string language (see "Notes" section below). Must consist of full days. closed : {'both', 'left', 'right', 'none'} Define which sides of the range are closed (inclusive). 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(Date)`. See Also -------- date_range datetime_ranges Notes ----- `interval` is created according to the following string language: - 1d (1 calendar day) - 1w (1 calendar week) - 1mo (1 calendar month) - 1q (1 calendar quarter) - 1y (1 calendar year) Or combine them: "1w2d" # 1 week, 2 days By "calendar day", we mean the corresponding time on the next day (which may not be 24 hours, due to daylight savings). Similarly for "calendar week", "calendar month", "calendar quarter", and "calendar year". Examples -------- >>> from datetime import date >>> df = pl.DataFrame( ... { ... "start": [date(2022, 1, 1), date(2022, 1, 2)], ... "end": date(2022, 1, 3), ... } ... ) >>> with pl.Config(fmt_str_lengths=50): ... df.with_columns(date_range=pl.date_ranges("start", "end")) shape: (2, 3) ┌────────────┬────────────┬──────────────────────────────────────┐ │ start ┆ end ┆ date_range │ │ --- ┆ --- ┆ --- │ │ date ┆ date ┆ list[date] │ ╞════════════╪════════════╪══════════════════════════════════════╡ │ 2022-01-01 ┆ 2022-01-03 ┆ [2022-01-01, 2022-01-02, 2022-01-03] │ │ 2022-01-02 ┆ 2022-01-03 ┆ [2022-01-02, 2022-01-03] │ └────────────┴────────────┴──────────────────────────────────────┘ )r rrr-r6r.r/r0r1s rr6r6srV'x00H(//L&s++J s|Z6RR S SF ,x))+++ Mr").) rrrrrrrrrrrr) rrrrrrrrrr#rr) rrrrrrrrrr&rr')r*) __future__r contextlibtypingrrpolarsrr.polars._utils.parserpolars._utils.wraprpolars.functions.range._utilsr suppress ImportError polars.polarsr-r r r r rrpolars._typingrrr r6rr"rrEs>""""""********!!!!!!555555((((((AAAAAAZ%%                 >2222222222########======== !$ !   !$ !   !$ !  !%@ $ @@@@@@F !$ !   !$ !   !$ !  !%T $ TTTTTTTTsAAA