Pick several columns

Selecting data · I07 · 8 MIN

Keep a two-dimensional table of selected columns.

Exercises within this concept

FollowFollow the techniqueCurrent exercise
ChangeAdapt a requirement
TransferChoose and combine

Understand the idea

A list of column names keeps a DataFrame, even when the list has just one name. The list also sets the column order.

Illustration · not the exercise output

A small example

df[["age", "name"]] shows age first and name second, with every original row.

Follow the code

Apply the idea to the supplied table. Read from top to bottom; the final line displays the result.

df[["rating", "price"]]

What each part does

df[["rating", "price"]]: Keep only rating and price, in that order, as a DataFrame.
inner [ ]: The inner brackets make a Python list of column names.
comma: Separate the two names inside that list.

Your inputs

The editable setup on the right creates df. Run executes the setup and your work from top to bottom.

Candy shop · 6 synthetic rows
candy	flavour	price	rating	shelf
Gummy Bear	fruity	1.2	4.1	A
Choco Pop	chocolate	2.1	4.6	B
Mint Bite	mint	1.5	3.8	A
Berry Loop	fruity	2.8	4.4	B
Cocoa Cube	chocolate	3.4	4.9	A
Lemon Drop	fruity	1.8	4	B

Your task · Follow

Using df, return a DataFrame containing rating, then price, in that order.

Hint

The inner brackets make a list; its order controls the output columns.

Reveal solution

One way to do it. Keep any supplied setup in the editor and use this in the Your work section.

df[["rating", "price"]]

Your task · Follow

Using df, return a DataFrame containing rating, then price, in that order.

YOUR PYTHON

Tab: indent · Shift+Tab: outdent · Esc, then Tab: leave editor

# Supplied setup — editable
import pandas as pd

data = {
    "candy": [
        "Gummy Bear",
        "Choco Pop",
        "Mint Bite",
        "Berry Loop",
        "Cocoa Cube",
        "Lemon Drop",
    ],
    "flavour": ["fruity", "chocolate", "mint", "fruity", "chocolate", "fruity"],
    "price": [1.2, 2.1, 1.5, 2.8, 3.4, 1.8],
    "rating": [4.1, 4.6, 3.8, 4.4, 4.9, 4],
    "shelf": ["A", "B", "A", "B", "A", "B"],
}

df = pd.DataFrame(data)

# Your work
df[[____, ____]]

Each run executes all editor code in a fresh Python session. Display a value by leaving it on the final line.

Python starts when you open a lesson.

Output

Run your code to see what Python returns.