主页 / coding / 问题 / 78689912
Accepted
bzm3r
Asked: 2024-07-01 06:09:56 +0800 CST

极点：将枚举值系列“分解”为枚举索引和枚举值的两列

  • 772
import polars as pl

dates = ["2024 Jan", "2024 Feb"]
DateEnum = pl.Enum(dates)

date_table = pl.DataFrame(
    {"date": pl.Series(raw_dates, dtype=DateEnum)}
)
date_table.explode("date") # fails

    {
        "date": pl.Series(raw_dates, dtype=plDateEnum).cast(dtype=str),
        "index": pl.Series(raw_dates, dtype=plDateEnum).cast(dtype=int),
    }
)
date_table # fails

我希望看到像这样的数据框：

from typing import Enum

pyDateEnum = Enum("Dates", raw_dates)
date_table = pl.DataFrame({
  "date": pl.Series("dates", [x.name for x in pyDateEnum], dtype=str),
  "index": pl.Series("indices", [x.value - 1 for x in pyDateEnum], dtype=int)
})
date_table

shape: (2, 2)
┌──────────┬───────┐
│ date     ┆ index │
│ ---      ┆ ---   │
│ str      ┆ i64   │
╞══════════╪═══════╡
│ 2024 Jan ┆ 0     │
│ 2024 Feb ┆ 1     │
└──────────┴───────┘

如何使用 Polars 来实现这一点，而不使用 Python 枚举？

python

1 个回答

  1. Best Answer

    我不确定是否有更本土化的方式来做到这一点，但你总是可以cast：

    date_table.with_columns(
    name = pl.col.date.cast(str),
    value = pl.col.date.cast(pl.UInt32)
)

┌──────────┬──────────┬───────┐
│ date     ┆ name     ┆ value │
│ ---      ┆ ---      ┆ ---   │
│ enum     ┆ str      ┆ u32   │
╞══════════╪══════════╪═══════╡
│ 2024 Jan ┆ 2024 Jan ┆ 0     │
│ 2024 Feb ┆ 2024 Feb ┆ 1     │
└──────────┴──────────┴───────┘
    • 1

相关问题