主页 / coding / 问题 / 78432213
Accepted
apostofes
Asked: 2024-05-05 20:37:04 +0800 CST

获取极坐标中时间序列 df 中的第一个、最后一个出现位置

  • 772

如何获取时间序列 df 中特定列的第一次、最后一次出现 (>0)，

shape: (4, 3)
┌────────────┬────────────┬────────────┐
│ date       ┆ column_one ┆ column_two │
│ ---        ┆ ---        ┆ ---        │
│ date       ┆ f64        ┆ i64        │
╞════════════╪════════════╪════════════╡
│ 2024-06-01 ┆ 0.0        ┆ 0          │
│ 2024-06-02 ┆ 0.0        ┆ 1          │
│ 2024-06-03 ┆ 1.0        ┆ 2          │
│ 2024-06-04 ┆ 1.2        ┆ 3          │
└────────────┴────────────┴────────────┘

并将其存储在单独的 df 中，

columns | first_appearance | last_appearance
column_one | 2024-06-03 | 2024-06-04
column_two | 2024-06-02 | 2024-06-04
python-polars

1 个回答

  1. Best Answer

    首先使用 获得具有列、、 和 的pl.DataFrame.melt长框架，即可获得所需的结果。然后，分组依据给出每个时间序列中第一个和最后一个出现 > 0 的值。datecolumnsvaluecolumns

    (
    df
    .melt(id_vars="date", variable_name="columns")
    .group_by("columns")
    .agg(
        pl.col("date").filter(pl.col("value") > 0).min().alias("first_appearance"),
        pl.col("date").filter(pl.col("value") > 0).max().alias("last_appearance"),
    )
)

    shape: (2, 3)
┌────────────┬──────────────────┬─────────────────┐
│ columns    ┆ first_appearance ┆ last_appearance │
│ ---        ┆ ---              ┆ ---             │
│ str        ┆ date             ┆ date            │
╞════════════╪══════════════════╪═════════════════╡
│ column_one ┆ 2024-06-03       ┆ 2024-06-04      │
│ column_two ┆ 2024-06-02       ┆ 2024-06-04      │
└────────────┴──────────────────┴─────────────────┘

    笔记。如果数据帧按 排序date，则可以使用pl.Expr.first/pl.Expr.last代替pl.Expr.min/ pl.Expr.max。pl.DataFrame.filter此外，聚合中的两个过滤器表达式可以在聚合之前替换为单个过滤器表达式。

    • 3

相关问题