串接多個條件
你也可以把多個條件串在一起。產品團隊想根據尺寸將產品分成「Large」、「Medium」或「Small」,以利最佳化出貨。
本練習屬於課程
使用 Polars 進行資料轉換
練習說明
- 檢查
"size"欄位是否等於"100ml",是的話回傳"Large"。 - 檢查
"size"欄位是否等於"50ml",是的話回傳"Medium"。 - 其他所有尺寸一律回傳
"Small",並將欄位命名為"size_category"。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
reviews.with_columns(
# Check if size is "100ml" and return "Large"
pl.when(pl.col("____") == "____").then(pl.lit("____"))
# Check if size is "50ml" and return "Medium"
.____(pl.col("size") == "____").then(pl.lit("____"))
# Return "Small" for other sizes
.____(pl.lit("____"))
.alias("size_category")
)