主页 / coding / 问题 / 79121591
Accepted
0x26res
Asked: 2024-10-24 19:01:38 +0800 CST

极坐标中的 pandas.Series.dt.ceil 的等价物

  • 772

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

例如：

  • 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),
    ]
    }
)
时间戳
2023-01-01T10：05：00.000000
2023-01-01T10：05：30.000000
2023-01-01T10：06：00.000000
2023-01-01T10：06：01.000000 
df['timestamp'].to_pandas().dt.ceil('1min')
时间戳
2023-01-01T10：05：00.000000
2023-01-01T10：06：00.000000
2023-01-01T10：06：00.000000
2023-01-01T10：07：00.000000

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

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

1 个回答

  1. Best Answer

    你可以用polars-xdt这个

    安装：

    pip install polars-xdt

    用法：

    import polars_xdt as xdt

df.with_columns(timestamp_ceil = xdt.ceil('timestamp', '1m'))

    shape: (4, 2)
┌─────────────────────┬─────────────────────┐
│ timestamp           ┆ timestamp_ceil      │
│ ---                 ┆ ---                 │
│ datetime[μs]        ┆ datetime[μs]        │
╞═════════════════════╪═════════════════════╡
│ 2023-01-01 10:05:00 ┆ 2023-01-01 10:05:00 │
│ 2023-01-01 10:05:30 ┆ 2023-01-01 10:06:00 │
│ 2023-01-01 10:06:00 ┆ 2023-01-01 10:06:00 │
│ 2023-01-01 10:06:01 ┆ 2023-01-01 10:07:00 │
└─────────────────────┴─────────────────────┘
    • 2

相关问题