From e338459f666c034d9b60087c37dc9e4f6ca28802 Mon Sep 17 00:00:00 2001 From: Mary Brown Date: Mon, 15 Jun 2026 13:22:21 +0000 Subject: [PATCH] Add price band --- src/template_code_location/ops/ops.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/template_code_location/ops/ops.py b/src/template_code_location/ops/ops.py index 3a73b18..92ce9cf 100644 --- 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