Skip to learning content

DATA SCIENCE PYTHON PLAYGROUND

Data Foundations · A little practice goes a long way.

← Visualise lessonsTINY TABLES · REAL PYTHON · YOUR PACE
Explain the evidence · V25 · 8 MIN

Legends and palettes

Make category mappings readable without colour alone.

Exercises within this concept

  1. FollowFollow the techniqueCurrent exercise
  2. ChangeMake overlap and category order explicit
  3. TransferChoose and combine

Understand the idea

A legend explains how appearance maps to data. Use distinct shapes alongside colour so group identity remains readable without colour.

Concept sketch: legend maps category to colour AND shapeCategoryABlegend maps category to colour AND shape
Illustration · not the exercise output

A small example

With hue and style both mapped to group, each group gets a colour and a marker shape in the legend.

Follow the code

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

import matplotlib.pyplot as plt
import seaborn as sns

fig, ax = plt.subplots(figsize=(6, 4))
sns.scatterplot(
    data=df, x="hours", y="score", hue="club", style="club", palette="colorblind", ax=ax
)
ax.legend(title="Category")
ax.set(title="Study club", xlabel="hours", ylabel="score")
fig.tight_layout()
plt.show()

What each part does

palette="colorblind"
a palette designed for categorical separation
ax.legend(title=...)
label the category key
style=
another cue alongside hue
Other choices for later exercises
hue_order=[...]
consistent category order across charts
alpha=0.6
partial transparency reveals overlapping points

Your inputs

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

Study club · 8 synthetic rows
studentclubhoursscoregroup
AriArt152A
BoCode371A
CyArt265B
DeeCode482B
EliArt589A
FloCode376A
GusArt693B
HanCode261B

Your task · Follow

  1. Using df, plot hours against score with sns.scatterplot, hue=club, style=club and palette="colorblind".
  2. Set the legend title to "Category".
  3. Chart: title "Study club"; x "hours"; y "score".
Hint

A legend describes the mapped category; redundant shapes help when colour is insufficient.

Reveal solution

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

import matplotlib.pyplot as plt
import seaborn as sns

fig, ax = plt.subplots(figsize=(6, 4))
sns.scatterplot(
    data=df, x="hours", y="score", hue="club", style="club", palette="colorblind", ax=ax
)
ax.legend(title="Category")
ax.set(title="Study club", xlabel="hours", ylabel="score")
fig.tight_layout()
plt.show()
Your task · Follow
  1. Using df, plot hours against score with sns.scatterplot, hue=club, style=club and palette="colorblind".
  2. Set the legend title to "Category".
  3. Chart: title "Study club"; x "hours"; y "score".

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

Edit Python. Control or Command plus Enter runs it. Tab indents by four spaces. Shift plus Tab outdents. Press Escape, then Tab or Shift plus Tab to leave the editor.

Each run executes all editor code in a fresh Python session. Display charts with plt.show().

Python starts when you open a lesson.

Output

Run your code to see what Python returns.