Add price band
This commit is contained in:
@@ -32,6 +32,9 @@ def transform_example_dataset(df: pd.DataFrame) -> pd.DataFrame:
|
|||||||
"""Applies a simple transformation to the example dataset."""
|
"""Applies a simple transformation to the example dataset."""
|
||||||
|
|
||||||
df["total_price"] = df["quantity"] * df["unit_price"]
|
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
|
return df
|
||||||
|
|||||||
Reference in New Issue
Block a user