创建 Object dtype
为了进行一次临时的文本探索,团队希望将每部电影的标题转换为由小写单词组成的 Python 集合。由于集合不是原生的 Polars dtype,请将其存储在 pl.Object 类型的列中。
已提供 DataFrame movies,且辅助函数 title_to_word_set(title) 已定义,会从标题中返回由小写单词组成的 Python 集合。
本练习是课程的一部分
使用 Polars 扩展与优化数据流水线
练习说明
- 对每个
movie_title运行title_to_word_set,并将返回的 Python 集合存入新列。 - 将新列命名为别名
title_word_set。
交互式实操练习
通过完成这段示例代码来试试这个练习。
result = movies.with_columns(
pl.col("movie_title")
# Convert each title to a set of words
.____(title_to_word_set, return_dtype=pl.____)
.alias("____")
).select("movie_title", "title_word_set").head(8)
print(result)