你可以
polars.Expr.cast
将物理整数数组转换为
polars.datatypes.Enum
以如下方式表示枚举的类别:
import polars as pl
enum = pl.Enum([*'abcd'])
df = pl.DataFrame({
'physical': [0, 0, 1, 2, 0, 3],
})
print(
df.with_columns(
categories=pl.col('physical').cast(enum)
)
)
# shape: (6, 2)
# ââââââââââââ¬âââââââââââââ
# â physical â categories â
# â --- â --- â
# â i64 â enum â
# ââââââââââââªâââââââââââââ¡
# â 0 â a â
# â 0 â a â
# â 1 â b â
# â 2 â c â
# â 0 â a â
# â 3 â d â
# ââââââââââââ´âââââââââââââ