開始使用免費開始

擷取主要類型

你剛加入一家串流電影新創的資料團隊。他們把電影中繼資料存成 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)
編輯並執行程式碼