创建 Enum 列
电影的状态有一套固定的词汇,例如 Released、Rumored、Post Production 等。由于允许值是预先已知的,用 Enum 比 Categorical 更合适:它会增加校验,在未知状态污染数据管道之前将其捕获。
movies 仍然可用,同时提供了预定义的 status_enum,其中列出了所有允许的取值。
本练习是课程的一部分
使用 Polars 扩展与优化数据流水线
练习说明
- 使用预定义的
status_enum将status列进行类型转换。
交互式实操练习
通过完成这段示例代码来试试这个练习。
result = movies.with_columns(
# Cast to the enum
pl.col("____").____(____)
).select("movie_title", "status").head(8)
print(result)