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.

This exercise is part of the course

Data Transformation with Polars

View Course

Exercise instructions

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

Hands-on interactive exercise

Have a go at this exercise by completing this sample code.

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

Edit and Run Code