def add_revenue_flag(df): df["revenue_flag"] = np.where( (df["quantity"] * df["price"] > 500) & (df["category"] == "Electronics"), "High", "Low" ) return df df = df.pipe(add_revenue_flag)