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.

Cet exercice fait partie du cours

<cours>Data Transformation with Polars</cours>

Voir le cours

Instructions de l’exercice

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

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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

Modifier et exécuter le code