combine multiple integer columns into a single datetime column in polars in python

main.py
import polars as pl

# Create a DataFrame with three integer columns
df = pl.DataFrame({
    "year": [2022, 2023, 2024],
    "month": [1, 2, 3],
    "day": [15, 20, 25]
})

# Combine the columns into a single datetime column using to_datetime and combine methods
df = df.with_column(
    pl.col("year").to_datetime(year=True)
    .combine(pl.col("month").to_datetime(month=True))
    .combine(pl.col("day").to_datetime(day=True))
    .alias("datetime")
)

print(df)
466 chars
19 lines

main.py
shape: (3, 4)
┌───────┬───────┬─────┬─────────────┐
│ year  ┆ month ┆ day ┆ datetime    │
│ ---   ┆ ---   ┆ --- ┆ ---         │
│ i32   ┆ i32   ┆ i32 ┆ datetime    │
╞═══════╪═══════╪═════╪═════════════╡
│ 2022  ┆ 1     ┆ 15  ┆ 2022-01-15  │
├───────┼───────┼─────┼─────────────┤
│ 2023  ┆ 2     ┆ 20  ┆ 2023-02-20  │
├───────┼───────┼─────┼─────────────┤
│ 2024  ┆ 3     ┆ 25  ┆ 2024-03-25  │
└───────┴───────┴─────┴─────────────┘
432 chars
13 lines