1. Lighthouse

Level 4 · Orders by channel and status

The request

From Angela Rivera, Head of Customer Support

“I suspect one channel has far more cancellations and refunds than the others. Can you count orders for every combination of channel and status?”

Your goal

Set result to a DataFrame with columns channel, status, n_orders (number of orders in that combination), one row per combination that occurs, sorted by channel then status A to Z, index reset.

  • Use the orders DataFramenot done yet
  • Show exactly channel status n_ordersnot done yet
  • 12 rows, sorted as askednot done yet

Your code

Python

Loading the editor…

    ResultPreview of orders · press Run to see yours
    Preview of orders
    order_idcustomer_idorder_datestatuschannel
    10011252023-01-01 20:51:19completedweb
    10021562023-01-02 07:58:46completedweb
    1003222023-01-02 10:25:27completedmarketplace
    1004482023-01-02 11:21:51completedmarketplace

    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.

    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
    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
    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
    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

    Grouping on more than one key

    Pass a list of columns and pandas makes a group for every combination that appears in the data:

    orders.groupby(["channel", "status"]).size()
    channel      status
    app          cancelled     ...
                 completed     ...
                 pending       ...
                 refunded      ...
    marketplace  cancelled     ...
    ...
    dtype: int64

    The result has a MultiIndex: two levels of row labels, channel then status. It’s sorted by both. Only combinations that actually occur appear; if no marketplace order were pending, there would be no row for it.

    From MultiIndex to a flat table

    A two-level index is awkward to read and merge. reset_index moves both levels into columns. On a Series, name= sets the name of the values column:

    orders.groupby(["channel", "status"]).size().reset_index(name="n_orders")

    The named-aggregation route gives the same table:

    orders.groupby(["channel", "status"], as_index=False).agg(n_orders=("order_id", "count"))

    Comparing groups fairly

    Raw counts mostly reflect channel size: web has the most orders, so it has the most of everything. To compare cancellation rates, divide by each channel’s total. normalize=True in value_counts does exactly this per group:

    orders.groupby("channel")["status"].value_counts(normalize=True)

    Each channel’s shares now add up to 1, and you can see whether one channel cancels more as a proportion, which is the question support is really asking.

    Reshaping into a grid with unstack

    People read a grid more easily than a long list. unstack moves the inner index level (status) into columns:

    orders.groupby(["channel", "status"]).size().unstack(fill_value=0)
    channel cancelled completed pending refunded
    app … … … …
    marketplace … … … …
    web … … … …

    fill_value=0 writes 0 for combinations that don’t occur, instead of NaN. pd.crosstab(orders["channel"], orders["status"]) builds the same grid directly, and pivot_table does it for any aggregation:

    orders.pivot_table(index="channel", columns="status", values="shipping_fee", aggfunc="mean")

    Use the long format (one row per combination) for further computation and merging, and the grid for showing people.

    Common mistakes

    • Using .count() on the whole grouped DataFrame. It returns one count column per original column, each skipping that column’s missing values, so coupon_code shows a much smaller number than order_id.
    • Forgetting name= in reset_index, which gives a values column called 0.
    • Drawing conclusions from raw counts across groups of very different sizes.

    If you know SQL

    This is GROUP BY channel, status with COUNT(*). unstack is a pivot, which SQL does with CASE WHEN inside SUM or a PIVOT clause.