1. Lighthouse

Level 1 · Add product names to order lines

The request

From Angela Rivera, Head of Customer Support

“Our agents see product IDs on order lines and have to look up every one. Can you give me the lines of our first ten orders, 1001 to 1010, with the product name and category attached?”

Your goal

Set result to a DataFrame with one row per order line of orders 1001 to 1010 (inclusive): columns order_item_id, order_id, name, category, quantity (name and category come from products, matched on product_id), sorted by order_item_id ascending, index reset.

  • Use the products DataFramenot done yet
  • Use the order_items DataFramenot done yet
  • Show exactly order_item_id order_id name category quantitynot done yet

Your code

Python

Loading the editor…

    ResultPreview of products · press Run to see yours
    Preview of products
    product_idnamecategorysubcategoryunit_price
    101Vale Velvet LoveseatFurnitureSofas1659.00
    102Alder Oak Coffee TableFurnitureTables919.00
    103Orchard Rattan Dining ChairFurnitureChairs789.00
    104Upland Oak BookcaseFurnitureStorage429.00

    Mote checks your result

    Press Run. Each row that matches the goal becomes a firefly in your lantern.

    DataFrames

    Loaded for you as pandas DataFrames, plus pd and np.

    products120 rows · 8 columns
    • product_idinteger
    • namevarchar
    • categoryvarchar
    • subcategoryvarchar
    • unit_pricedecimal
    • unit_costdecimal
    • launched_ondate
    • is_activeboolean
    First 5 rows of products
    product_idnamecategorysubcategoryunit_priceunit_costlaunched_onis_active
    101Vale Velvet LoveseatFurnitureSofas1659.00NaN2023-09-11true
    102Alder Oak Coffee TableFurnitureTables919.00503.502024-05-16true
    103Orchard Rattan Dining ChairFurnitureChairs789.00429.872023-05-27true
    104Upland Oak BookcaseFurnitureStorage429.00214.702022-12-20true
    105Avery Leather Bed FrameFurnitureBeds449.00232.702021-08-31true
    order_items30,338 rows · 6 columns
    • order_item_idinteger
    • order_idinteger
    • product_idinteger
    • quantityinteger
    • unit_pricedecimal
    • discount_pctdecimal
    First 5 rows of order_items
    order_item_idorder_idproduct_idquantityunit_pricediscount_pct
    11001121931.990.00
    21001154311.990.00
    31002154411.990.00
    410021941149.000.00
    51002193219.990.00
    customers2,018 rows · 9 columns
    • customer_idinteger
    • first_namevarchar
    • last_namevarchar
    • emailvarchar
    • cityvarchar
    • statevarchar
    • signup_datedate
    • segmentvarchar
    • referral_sourcevarchar
    First 5 rows of customers
    customer_idfirst_namelast_nameemailcitystatesignup_datesegmentreferral_source
    1OwenHudsonowen.hudson@icloud.comSan FranciscoCA2022-06-02consumersearch
    2DouglasMendozadmendoza69@gmail.comAtlantaGA2022-06-02consumerNaN
    3HenryBrookshbrooks@lakesidecafe.comLos AngelesCA2022-06-04businessNaN
    4BrianGrahambgraham@bluecedarhomes.comSeattleWA2022-06-06businesspinterest
    5KyleJimenezkjimenez27@yahoo.comIndianapolisIN2022-06-07consumerNaN
    orders12,000 rows · 7 columns
    • order_idinteger
    • customer_idinteger
    • order_datetimestamp
    • statusvarchar
    • channelvarchar
    • shipping_feedecimal
    • coupon_codevarchar
    First 5 rows of orders
    order_idcustomer_idorder_datestatuschannelshipping_feecoupon_code
    10011252023-01-01 20:51:19completedweb0.00NaN
    10021562023-01-02 07:58:46completedweb0.00NaN
    1003222023-01-02 10:25:27completedmarketplace4.99WELCOME10
    1004482023-01-02 11:21:51completedmarketplace4.99WELCOME10
    1005822023-01-02 11:43:36completedweb0.00WELCOME10
    returns1,673 rows · 5 columns
    • return_idinteger
    • order_item_idinteger
    • returned_ondate
    • reasonvarchar
    • refund_amountdecimal
    First 5 rows of returns
    return_idorder_item_idreturned_onreasonrefund_amount
    1142023-01-10damaged83.69
    2422023-01-10wrong_item19.99
    31052023-01-14wrong_item10.99
    4242023-01-16changed_mind26.99
    51262023-01-16late20.99
    employees60 rows · 8 columns
    • employee_idinteger
    • first_namevarchar
    • last_namevarchar
    • departmentvarchar
    • titlevarchar
    • manager_idinteger
    • hire_datedate
    • salaryinteger
    First 5 rows of employees
    employee_idfirst_namelast_namedepartmenttitlemanager_idhire_datesalary
    1CatherineHaleExecutiveChief Executive OfficerNaN2016-03-01285000
    2RobertKimExecutiveChief Operating Officer12017-01-09235000
    3MeganCollinsSalesHead of Sales12018-04-16172000
    4ChloeGutierrezSalesAccount Executive32020-01-2172000
    5FrankBellSalesAccount Executive32023-05-2173000
    web_sessions40,000 rows · 7 columns
    • session_idinteger
    • customer_idinteger
    • started_attimestamp
    • devicevarchar
    • sourcevarchar
    • pages_viewedinteger
    • convertedboolean
    First 5 rows of web_sessions
    session_idcustomer_idstarted_atdevicesourcepages_viewedconverted
    1NaN2023-01-01 23:02:24mobilepaid1false
    2182023-01-01 15:07:49desktoppaid1false
    3NaN2023-01-01 16:55:46desktopdirect7false
    4NaN2023-01-01 12:43:28mobileorganic3false
    5NaN2023-01-01 13:06:21mobilesocial2false
    marketing_spend3,276 rows · 3 columns
    • spend_datedate
    • channelvarchar
    • amountdecimal
    First 5 rows of marketing_spend
    spend_datechannelamount
    2023-01-01paid492.99
    2023-01-01email37.38
    2023-01-01social200.53
    2023-01-02paid709.13
    2023-01-02email64.13

    The lesson

    Why tables are split

    order_items stores a product_id on each line instead of the product’s name, category and price. That’s deliberate: the product’s details live in one place, products, so renaming a product means changing one row, not 30,000. The cost is that you have to join the tables back together when you need both.

    merge

    order_items.merge(products, on="product_id")

    For each order line, pandas finds the product row with the same product_id and puts their columns side by side. The column you match on is the key. If the key has different names in the two tables, use left_on= and right_on=. For example, each employee’s manager_id points at another employee’s employee_id, so this puts every employee next to their manager’s details:

    employees.merge(employees, left_on="manager_id", right_on="employee_id", suffixes=("", "_manager"))

    That gives 59 rows, not 60: the CEO has no manager, so the inner merge leaves that row out.

    The how= argument decides what happens to rows without a match:

    how keeps
    "inner" (default) only rows with a match in both tables
    "left" every row from the left table; NaN where the right has no match
    "right" every row from the right table
    "outer" everything from both

    Every order line here has a valid product, so inner and left give the same result. Write how= explicitly anyway; it tells the reader you thought about it.

    Overlapping column names

    Both tables have a unit_price column, and they mean different things: the price paid on the line versus today’s list price. pandas keeps both and adds suffixes, unit_price_x and unit_price_y. Those names are easy to mix up, so set your own:

    order_items.merge(products, on="product_id", suffixes=("_paid", "_list"))

    Or select only the columns you need from the right table before merging, which also keeps the result narrow:

    order_items.merge(products[["product_id", "name", "category"]], on="product_id")

    Check the row count

    The single most useful habit with merges: compare row counts before and after.

    len(order_items)                                   # lines before
    len(order_items.merge(products, on="product_id"))  # should be the same

    Each line has exactly one product, so the count must not change. If it grew, the right table had duplicate keys and some lines were copied. If it shrank, some lines had no match and an inner join dropped them. You’ll see both problems later in this chapter.

    pandas can check this for you. validate="many_to_one" says “many lines may share a product, but each product appears once in products”, and raises an error if that’s not true:

    order_items.merge(products, on="product_id", how="inner", validate="many_to_one")

    Other options are "one_to_one" and "one_to_many". Adding validate costs nothing and turns a silent mistake into a loud one.

    Common mistakes

    • Merging without on=. pandas then joins on every column name the tables share, here product_id and unit_price, and most lines fail to match because prices changed.
    • Using unit_price_x later without remembering which table it came from.
    • Not checking the row count after a merge.

    If you know SQL

    This is SELECT ... FROM order_items JOIN products USING (product_id), and how= is the join type.