1. Lighthouse

Level 3 · Pick the columns you need

The request

From Logan Brooks, Email Marketing Specialist

“I'm setting up a new email audience. From the customers table I only need the customer ID, email, state and segment, in that order: no names, no signup dates. Send me the first 20 rows so I can check the format.”

Your goal

Set result to a DataFrame with the first 20 rows of customers and exactly the columns customer_id, email, state, segment, in that order. Keep the original index.

  • Use the customers DataFramenot done yet
  • Show exactly customer_id email state segmentnot done yet
  • 20 rows, sorted as askednot done yet

Your code

Python

Loading the editor…

    ResultPreview of customers · press Run to see yours
    Preview of customers
    customer_idfirst_namelast_nameemailcity
    1OwenHudsonowen.hudson@icloud.comSan Francisco
    2DouglasMendozadmendoza69@gmail.comAtlanta
    3HenryBrookshbrooks@lakesidecafe.comLos Angeles
    4BrianGrahambgraham@bluecedarhomes.comSeattle

    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.

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

    One column: a Series

    Square brackets with a column name give you that column:

    customers["email"]

    The result is a Series: one column of values plus the index. Series have their own useful methods, like .nunique(), .value_counts() or .str.lower(), which you’ll use all the time.

    You may also see customers.email (dot access). It works for simple names, but it breaks for names with spaces or names that clash with DataFrame methods (a column called count or size), and it can’t create new columns. Bracket access always works, so make it your habit.

    Several columns: a DataFrame

    To select more than one column, pass a list of names:

    customers[["customer_id", "email", "state", "segment"]]

    The double brackets confuse everyone at first. The outer brackets mean “select from this DataFrame”. The inner brackets are just Python’s list syntax. The same thing written in two steps makes this clearer:

    cols = ["customer_id", "email", "state", "segment"]
    customers[cols]

    The columns come back in the order you list them, so this is also how you reorder columns.

    One column as a DataFrame

    A list with a single name gives you a one-column DataFrame instead of a Series:

    customers["email"]       # Series
    customers[["email"]]     # DataFrame with one column

    Which one you want depends on what comes next. Series methods (value_counts, .str) want a Series; exporting a table or merging wants a DataFrame.

    Selecting columns does not copy the table

    The selection is a new object. customers itself still has all nine columns. If you want to keep the narrower table, assign it:

    contacts = customers[["customer_id", "email", "state", "segment"]]

    If you’re going to modify that new table (add a column, change values), make it an explicit copy: contacts = customers[[...]].copy(). That avoids pandas’ SettingWithCopyWarning and makes it clear the two tables are independent.

    Finding columns by pattern

    When a table has dozens of columns, filter helps:

    orders.filter(like="_id")    # every column whose name contains "_id"

    Why bother narrowing?

    Selecting only what you need makes outputs readable, makes exports smaller, and is a privacy habit worth having: a marketing audience doesn’t need signup dates, and a shared spreadsheet shouldn’t carry columns nobody asked for.

    Common mistakes

    • Single brackets around a list of names written without the inner list: customers["customer_id", "email"] raises KeyError, because pandas looks for one column named by that pair.
    • Typos and capitalisation: customers["Email"] raises KeyError: 'Email'.
    • Expecting customers[["email"]] to be a Series and calling .str on it.

    If you know SQL

    customers[["customer_id", "email"]] is SELECT customer_id, email FROM customers.