The request

From Priya Shah, Head of Data
Your goal
Using per_product from the starter (columns unit_price, units_sold, revenue, return_rate, margin, in that order), set result to a DataFrame with the 3 pairs of different columns that have the largest absolute correlation, strongest first: columns var_1, var_2 (the pair, var_1 being the one that comes first in per_product) and r (the correlation with its sign, rounded to 3), index reset.
- Use the
productsDataFramenot done yet - Use the
ordersDataFramenot done yet - Show exactly
var_1var_2rnot done yet
Orange is a positive correlation, blue a negative one; the stronger the colour, the closer r is to +1 or -1. The diagonal is always 1: every column matches itself.
Show the numbers
| unit_price | units_sold | revenue | return_rate | margin | |
|---|---|---|---|---|---|
| unit_price | 1.00 | -0.43 | 0.38 | 0.16 | -0.23 |
| units_sold | -0.43 | 1.00 | 0.14 | -0.05 | 0.10 |
| revenue | 0.38 | 0.14 | 1.00 | 0.11 | -0.17 |
| return_rate | 0.16 | -0.05 | 0.11 | 1.00 | 0.04 |
| margin | -0.23 | 0.10 | -0.17 | 0.04 | 1.00 |
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 Priya 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 |
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 |
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 |
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
One call, every pair
Level 1 measured one pair. When a table has several numeric columns, DataFrame.corr() measures
all of them at once and returns a square correlation matrix: one row and one column per
variable, r in every cell.
per_product.corr().round(2)
Three things are always true of it. The diagonal is 1 (each column against itself). The matrix is
symmetric: the cell for price and units equals the cell for units and price. And pandas uses the
rows where both columns have values, so the 9 products without a cost still count in every pair
that does not involve margin.
Reading it as a heatmap
Twenty-five numbers are hard to scan, so the matrix is usually drawn as a heatmap: one coloured cell per pair, the number printed inside. Two rules make a correlation heatmap honest:
- A fixed scale from -1 to +1. If the colours stretch to the strongest value in this table, a weak 0.4 is painted as dark as a near-perfect 0.95 would be in another table.
- A diverging palette: one colour for positive, another for negative, pale near zero. Here orange is positive and blue is negative.
On the chart, the darkest off-diagonal cell is price against units sold, in blue.
Strong, moderate, weak
A common rule of thumb: |r| above 0.7 is strong, 0.3 to 0.7 moderate, below 0.3 weak. On Harbor & Pine’s products only two pairs are even moderate: price and units sold (-0.425) and price and revenue (0.375). Everything else is below 0.25, and return rate is barely related to anything.
Rank pairs by the absolute value, then report the sign: “price and units sold have the strongest relationship, a moderate negative one”.
The grain changes the answer
You might expect units sold and revenue to move together strongly. Across products they hardly do (r = 0.14), because a product that sells 900 cheap units can earn less than one that sells 50 sofas. Count the same two numbers per month and r is 0.95: busy months are busy in units and dollars. Per order it is 0.60. A correlation describes the rows you computed it on, so always say what one row is.
Common mistakes
- Sorting by r instead of |r|. Strong negative pairs end up at the bottom.
- Counting pairs twice and keeping the diagonal. Use only the cells above the diagonal.
- Letting the colour scale float. Fix it at -1 to 1.
- Writing “causes”. A matrix shows which columns are associated; say “is associated with” or “tends to go with”, and check the scatter plot before anything else.
Lantern lit!
Practised: correlation matrixheatmapcorrelationassociation
- Solved
- No hints
- First run
