0x26res's questions

我正在尝试将时间戳四舍五入到极坐标中的下一分钟。

例如：

• 2023-01-01 10:05:00应该留下2023-01-01 10:05:00
• 2023-01-01 10:05:01应该是2023-01-01 10:06:00

这在带有 ceil 的 pandas 中有效：

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')

我在极地中发现的唯一方法如下：

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))
)