开始使用免费开始使用

创建 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)
编辑并运行代码