开始使用免费开始使用

创建 Enum 列

电影的状态有一套固定的词汇,例如 ReleasedRumoredPost Production 等。由于允许值是预先已知的,用 EnumCategorical 更合适:它会增加校验,在未知状态污染数据管道之前将其捕获。

movies 仍然可用,同时提供了预定义的 status_enum,其中列出了所有允许的取值。

本练习是课程的一部分

使用 Polars 扩展与优化数据流水线

查看课程

练习说明

  • 使用预定义的 status_enumstatus 列进行类型转换。

交互式实操练习

通过完成这段示例代码来试试这个练习。

result = movies.with_columns(
    # Cast to the enum
    pl.col("____").____(____)
).select("movie_title", "status").head(8)
print(result)
编辑并运行代码