The request

From Laura Bennett, Head of Finance
Your goal
The starter runs but fills missing unit_cost with 0. Fix it: compute cost_ratio = unit_cost / unit_price, fill missing cost_ratio values with the median cost_ratio of the product's category (groupby + transform('median')), and set margin_pct = 1 - cost_ratio. Set result to a DataFrame with columns category, n_products (products per category, all 120 kept) and avg_margin (mean margin_pct, rounded to 3), sorted by avg_margin descending then category ascending, index reset.
- Use the
productsDataFramenot done yet - Show exactly
categoryn_productsavg_marginnot done yet - 6 rows, sorted as askednot done yet
Your code
Loading the editor…
Reference solution
| product_id | name | category | subcategory | unit_price |
|---|---|---|---|---|
| 101 | Vale Velvet Loveseat | Furniture | Sofas | 1659.00 |
| 102 | Alder Oak Coffee Table | Furniture | Tables | 919.00 |
| 103 | Orchard Rattan Dining Chair | Furniture | Chairs | 789.00 |
| 104 | Upland Oak Bookcase | Furniture | Storage | 429.00 |
Mote checks your result
Press Run. Each row that matches the goal becomes a firefly in your lantern.
What Laura gets, and why
DataFrames
Loaded for you as pandas DataFrames, plus pd and np.
products120 rows · 8 columns
product_idintegernamevarcharcategoryvarcharsubcategoryvarcharunit_pricedecimalunit_costdecimallaunched_ondateis_activeboolean
| product_id | name | category | subcategory | unit_price | unit_cost | launched_on | is_active |
|---|---|---|---|---|---|---|---|
| 101 | Vale Velvet Loveseat | Furniture | Sofas | 1659.00 | NaN | 2023-09-11 | true |
| 102 | Alder Oak Coffee Table | Furniture | Tables | 919.00 | 503.50 | 2024-05-16 | true |
| 103 | Orchard Rattan Dining Chair | Furniture | Chairs | 789.00 | 429.87 | 2023-05-27 | true |
| 104 | Upland Oak Bookcase | Furniture | Storage | 429.00 | 214.70 | 2022-12-20 | true |
| 105 | Avery Leather Bed Frame | Furniture | Beds | 449.00 | 232.70 | 2021-08-31 | true |
customers2,018 rows · 9 columns
customer_idintegerfirst_namevarcharlast_namevarcharemailvarcharcityvarcharstatevarcharsignup_datedatesegmentvarcharreferral_sourcevarchar
| customer_id | first_name | last_name | city | state | signup_date | segment | referral_source | |
|---|---|---|---|---|---|---|---|---|
| 1 | Owen | Hudson | owen.hudson@icloud.com | San Francisco | CA | 2022-06-02 | consumer | search |
| 2 | Douglas | Mendoza | dmendoza69@gmail.com | Atlanta | GA | 2022-06-02 | consumer | NaN |
| 3 | Henry | Brooks | hbrooks@lakesidecafe.com | Los Angeles | CA | 2022-06-04 | business | NaN |
| 4 | Brian | Graham | bgraham@bluecedarhomes.com | Seattle | WA | 2022-06-06 | business | |
| 5 | Kyle | Jimenez | kjimenez27@yahoo.com | Indianapolis | IN | 2022-06-07 | consumer | NaN |
orders12,000 rows · 7 columns
order_idintegercustomer_idintegerorder_datetimestampstatusvarcharchannelvarcharshipping_feedecimalcoupon_codevarchar
| order_id | customer_id | order_date | status | channel | shipping_fee | coupon_code |
|---|---|---|---|---|---|---|
| 1001 | 125 | 2023-01-01 20:51:19 | completed | web | 0.00 | NaN |
| 1002 | 156 | 2023-01-02 07:58:46 | completed | web | 0.00 | NaN |
| 1003 | 22 | 2023-01-02 10:25:27 | completed | marketplace | 4.99 | WELCOME10 |
| 1004 | 48 | 2023-01-02 11:21:51 | completed | marketplace | 4.99 | WELCOME10 |
| 1005 | 82 | 2023-01-02 11:43:36 | completed | web | 0.00 | WELCOME10 |
order_items30,338 rows · 6 columns
order_item_idintegerorder_idintegerproduct_idintegerquantityintegerunit_pricedecimaldiscount_pctdecimal
| order_item_id | order_id | product_id | quantity | unit_price | discount_pct |
|---|---|---|---|---|---|
| 1 | 1001 | 121 | 9 | 31.99 | 0.00 |
| 2 | 1001 | 154 | 3 | 11.99 | 0.00 |
| 3 | 1002 | 154 | 4 | 11.99 | 0.00 |
| 4 | 1002 | 194 | 1 | 149.00 | 0.00 |
| 5 | 1002 | 193 | 2 | 19.99 | 0.00 |
returns1,673 rows · 5 columns
return_idintegerorder_item_idintegerreturned_ondatereasonvarcharrefund_amountdecimal
| return_id | order_item_id | returned_on | reason | refund_amount |
|---|---|---|---|---|
| 1 | 14 | 2023-01-10 | damaged | 83.69 |
| 2 | 42 | 2023-01-10 | wrong_item | 19.99 |
| 3 | 105 | 2023-01-14 | wrong_item | 10.99 |
| 4 | 24 | 2023-01-16 | changed_mind | 26.99 |
| 5 | 126 | 2023-01-16 | late | 20.99 |
employees60 rows · 8 columns
employee_idintegerfirst_namevarcharlast_namevarchardepartmentvarchartitlevarcharmanager_idintegerhire_datedatesalaryinteger
| employee_id | first_name | last_name | department | title | manager_id | hire_date | salary |
|---|---|---|---|---|---|---|---|
| 1 | Catherine | Hale | Executive | Chief Executive Officer | NaN | 2016-03-01 | 285000 |
| 2 | Robert | Kim | Executive | Chief Operating Officer | 1 | 2017-01-09 | 235000 |
| 3 | Megan | Collins | Sales | Head of Sales | 1 | 2018-04-16 | 172000 |
| 4 | Chloe | Gutierrez | Sales | Account Executive | 3 | 2020-01-21 | 72000 |
| 5 | Frank | Bell | Sales | Account Executive | 3 | 2023-05-21 | 73000 |
web_sessions40,000 rows · 7 columns
session_idintegercustomer_idintegerstarted_attimestampdevicevarcharsourcevarcharpages_viewedintegerconvertedboolean
| session_id | customer_id | started_at | device | source | pages_viewed | converted |
|---|---|---|---|---|---|---|
| 1 | NaN | 2023-01-01 23:02:24 | mobile | paid | 1 | false |
| 2 | 18 | 2023-01-01 15:07:49 | desktop | paid | 1 | false |
| 3 | NaN | 2023-01-01 16:55:46 | desktop | direct | 7 | false |
| 4 | NaN | 2023-01-01 12:43:28 | mobile | organic | 3 | false |
| 5 | NaN | 2023-01-01 13:06:21 | mobile | social | 2 | false |
marketing_spend3,276 rows · 3 columns
spend_datedatechannelvarcharamountdecimal
| spend_date | channel | amount |
|---|---|---|
| 2023-01-01 | paid | 492.99 |
| 2023-01-01 | 37.38 | |
| 2023-01-01 | social | 200.53 |
| 2023-01-02 | paid | 709.13 |
| 2023-01-02 | 64.13 |
The lesson
Why 0 is the wrong fill
Nine products have no unit_cost (NaN). A missing cost means “we don’t know”, not “free”.
fillna(0) turns “we don’t know” into “it costs nothing”, and the margin formula then gives those
products 100%. Five of the nine are Furniture (the Vale Velvet Loveseat, Tidewater Teak Loveseat,
Weston Ash Bed Frame, Alder Linen Sideboard and Sable Oak Daybed), so Furniture’s average jumped to
65.8% and to the top of the list. Laura knew the business well enough to see that was impossible.
That’s the best kind of check.
Fill with the group’s median: transform
Laura asked for an estimate from similar products. Costs differ hugely in dollars, so fill the cost ratio (cost as a share of price), which is comparable across products in a category:
products["cost_ratio"] = products["unit_cost"] / products["unit_price"]
products.groupby("category")["cost_ratio"].median()
# Furniture 0.476, Outdoor 0.445, Lighting 0.437, Decor 0.434, Kitchen 0.428, Textiles 0.416
agg or median() after a groupby returns one row per category. To fill a column you need the
category’s median on every product row, lined up with the table. That’s what transform does: it
computes per group and broadcasts the answer back to every row of the group, keeping the original
index.
category_median = products.groupby("category")["cost_ratio"].transform("median")
products["cost_ratio"] = products["cost_ratio"].fillna(category_median)
fillna with a Series fills each gap with the value at the same index label, so every loveseat gets
the Furniture median and every grill the Outdoor one. The median ignores the NaNs it’s replacing.
The corrected report
Kitchen now leads at 57.5% and Furniture is last at 53.9%, which matches what finance sees. All 120 products stay in the report, 20 per category. Simply dropping the nine would also give a sensible ranking, but with only 15 Furniture products, and Laura asked to keep them.
Lighting (55.8%) and Outdoor (55.7%) are close. Before rounding they’re far enough apart, but the category tie-breaker in the sort keeps the order fixed whatever the data does next month.
Common mistakes
fillna(0)on a cost, price or salary. Zero is a real value, not “unknown”.- Using
.agg("median")or.median()instead oftransform, then trying to fill with a table that has six rows. - Filling with the median of the whole column, which ignores how categories differ.
- Filling
unit_costwith the category’s median cost in dollars, which gives the $259 Alder Linen Sideboard the typical cost of a far pricier sofa or bed.
Lantern lit!
Practised: fillnatransformgroupbymediannandebugging
- Solved
- No hints
- First run
