Chaining multiple conditions

You can also chain multiple conditions together. The product team wants to categorize products as "Large", "Medium", or "Small" based on their size for shipping optimization.

Deze oefening maakt deel uit van de cursus

Data Transformation with Polars

Bekijk cursus

Oefeninstructies

Check if the "size" column equals "100ml" and return "Large".
Check if the "size" column equals "50ml" and return "Medium".
Return "Small" for all other sizes and name the column "size_category".

Interactieve oefening met praktijkervaring

Probeer deze oefening door deze voorbeeldcode aan te vullen.

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")
)

Code bewerken en uitvoeren