0x26res's questions

Estou tentando arredondar o registro de data e hora para os próximos minutos em polares.

Por exemplo:

• 2023-01-01 10:05:00deveria ficar2023-01-01 10:05:00
• 2023-01-01 10:05:01deveria ser2023-01-01 10:06:00

Isso funciona em pandas com teto:

import polars as pl
import datetime

df = pl.DataFrame(
    {'timestamp' :[
        datetime.datetime(2023, 1, 1, 10, 5, 0),
        datetime.datetime(2023, 1, 1, 10, 5, 30),
        datetime.datetime(2023, 1, 1, 10, 6, 0),
        datetime.datetime(2023, 1, 1, 10, 6, 1),
    ]
    }
)

df['timestamp'].to_pandas().dt.ceil('1min')

A única maneira que encontrei em polares é a seguinte:

df.with_columns(
    pl.when(pl.col('timestamp').dt.truncate('1m') == pl.col('timestamp'))
    .then(pl.col('timestamp'))
    .otherwise(pl.col('timestamp').dt.truncate('1m') + datetime.timedelta(minutes=1))
)