Encode categories

Combining & preparing · W27 · 8 MIN

Represent category membership as indicator columns.

Exercises within this concept

FollowFollow the techniqueCurrent exercise
ChangeAdapt a requirement
TransferChoose and combine

Understand the idea

One-hot encoding replaces category labels with an indicator column for each label.

Illustration · not the exercise output

A small example

For species Cat and Dog: the Cat row has species_Cat=1 and species_Dog=0; the Dog row has the reverse.

Follow the code

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

pd.get_dummies(df, columns=["flavour"], dtype=int)

What each part does

pd.get_dummies(df, columns=[...]): encode selected columns
dtype=int: use 0 and 1 rather than False and True

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

One-hot encode flavour in df with integer indicator columns, preserving the other columns.
Return the encoded DataFrame.
Use: get_dummies().

Hint

Specify dtype=int for 0/1 indicators rather than Boolean values.

Reveal solution

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

pd.get_dummies(df, columns=["flavour"], dtype=int)

Your task · Follow

One-hot encode flavour in df with integer indicator columns, preserving the other columns.
Return the encoded DataFrame.
Use: get_dummies().

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
# Build the requested result from the given table
result = ____
result

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.