提取主类型
您刚加入一家流媒体电影初创公司的数据团队。他们将影片元数据存储在一个 Parquet 文件中，每部电影的 genres 都是字符串列表，因为大多数电影涵盖多个类型。请将每行中列出的第一个类型提取出来，作为一个干净的顶层列 primary_genre。
polars 已作为 pl 加载，DataFrame movies 已从 Parquet 文件预加载。
本练习是课程的一部分
使用 Polars 扩展与优化数据流水线
练习说明
- 使用列表表达式，从每行的
genres列表中提取第一个元素。 - 将新列命名为
primary_genre。
交互式实操练习
通过完成这段示例代码来试试这个练习。
result = movies.select(
"movie_title",
"genres",
# Get the first genre
pl.col("genres").list.____(0).alias("____"),
).head(8)
print(result)