擷取主要類型
你剛加入一家串流電影新創的資料團隊。他們把電影中繼資料存成 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)