add price band to data and calculate it

2026-06-15 13:12:02 +00:00
parent be3c271041
commit 9f65e5ed9d
1 changed files with 3 additions and 0 deletions
--- a/src/template_code_location/ops/ops.py
+++ b/src/template_code_location/ops/ops.py
@@ -33,5 +33,8 @@ def transform_example_dataset(df: pd.DataFrame) -> pd.DataFrame:
    
    df["total_price"] = df["quantity"] * df["unit_price"]

+    df["price_band"] = df["total_price"].apply(
+        lambda value: "high" if value >= 25 else "standard"
+    )
        
    return df