Categorical scatter with jitter in Matplotlib

Introduction

We use categorical scatter plots with jitter to show data points for categories clearly. Jitter adds small random shifts so points don't overlap.

When you want to see individual data points for categories like fruits or colors.

When many points overlap in a category and you want to spread them out to see density.

When comparing groups and you want to show all data points, not just summaries.

When you want a simple way to visualize distribution within categories.

Syntax

Matplotlib

import matplotlib.pyplot as plt
import numpy as np

categories = ['A', 'B', 'C']
values = [5, 7, 6, 8, 5, 7, 6, 9, 5]
category_labels = ['A', 'A', 'B', 'B', 'B', 'C', 'C', 'C', 'C']

# Convert categories to numbers
x = [categories.index(cat) for cat in category_labels]

# Add jitter
jitter = np.random.uniform(-0.1, 0.1, size=len(x))
x_jittered = x + jitter

plt.scatter(x_jittered, values)
plt.xticks(range(len(categories)), categories)
plt.xlabel('Category')
plt.ylabel('Value')
plt.title('Categorical scatter with jitter')
plt.show()

Jitter is a small random number added to category positions to avoid overlap.

Categories are converted to numbers because scatter needs numeric x values.

Examples

This example shows two categories with jitter to spread points horizontally.

Matplotlib

import matplotlib.pyplot as plt
import numpy as np

categories = ['X', 'Y']
values = [1, 2, 3, 4, 5, 6]
category_labels = ['X', 'X', 'Y', 'Y', 'Y', 'X']

x = [categories.index(cat) for cat in category_labels]
jitter = np.random.uniform(-0.05, 0.05, len(x))
x_jittered = x + jitter

plt.scatter(x_jittered, values)
plt.xticks(range(len(categories)), categories)
plt.show()

Here jitter is larger to spread points more for better visibility.

Matplotlib

import matplotlib.pyplot as plt
import numpy as np

cats = ['Dog', 'Cat', 'Bird']
vals = [3, 5, 2, 4, 6, 7, 3, 5]
labels = ['Dog', 'Dog', 'Cat', 'Cat', 'Bird', 'Bird', 'Bird', 'Dog']

x = [cats.index(c) for c in labels]
jitter = np.random.uniform(-0.2, 0.2, len(x))
x_jittered = x + jitter

plt.scatter(x_jittered, vals, color='green')
plt.xticks(range(len(cats)), cats)
plt.title('Pets values with jitter')
plt.show()

Sample Program

This program shows how to plot values for color categories with jitter to avoid overlapping points. The jitter is random but fixed by seed for consistent results.

Matplotlib

import matplotlib.pyplot as plt
import numpy as np

# Define categories and values
categories = ['Red', 'Blue', 'Green']
values = [10, 15, 10, 20, 25, 15, 10, 30, 20]
category_labels = ['Red', 'Red', 'Blue', 'Blue', 'Blue', 'Green', 'Green', 'Green', 'Green']

# Convert categories to numeric positions
x = [categories.index(cat) for cat in category_labels]

# Add jitter to x positions
np.random.seed(42)  # For reproducible jitter
jitter = np.random.uniform(-0.15, 0.15, size=len(x))
x_jittered = x + jitter

# Create scatter plot
plt.scatter(x_jittered, values, color='purple')
plt.xticks(range(len(categories)), categories)
plt.xlabel('Color')
plt.ylabel('Value')
plt.title('Categorical scatter plot with jitter')
plt.grid(True, linestyle='--', alpha=0.5)
plt.show()

Important Notes

Use a small jitter range to keep points near their category.

Setting a random seed helps get the same jitter every time you run the code.

Jitter only affects horizontal position for categorical scatter plots.

Summary

Categorical scatter plots show individual points for categories.

Jitter adds small random shifts to avoid overlapping points.

This helps visualize data distribution within categories clearly.

Practice

1. What is the main purpose of adding jitter in a categorical scatter plot using matplotlib?

easy

A. To increase the size of the points

B. To change the color of the points randomly

C. To spread overlapping points so they are easier to see

D. To remove outliers from the data

Solution

1. Step 1: Understand what jitter does

   Jitter adds small random noise to the position of points on the categorical axis.

2. Step 2: Effect of jitter on overlapping points

   This noise spreads points that would overlap, making each point visible.

3. Final Answer:

   To spread overlapping points so they are easier to see -> Option C

4. Quick Check:

   Jitter spreads points = A [OK]

Hint: Jitter spreads points to avoid overlap [OK]

Common Mistakes:

• Thinking jitter changes point color
• Assuming jitter increases point size
• Believing jitter removes data points

2. Which of the following matplotlib code snippets correctly adds jitter to a categorical scatter plot on the x-axis?

easy

A. plt.scatter(x + np.random.uniform(-0.1, 0.1, size=len(x)), y)

B. plt.scatter(x * np.random.uniform(-0.1, 0.1, size=len(x)), y)

C. plt.scatter(x - np.random.randint(1, 5, size=len(x)), y)

D. plt.scatter(x / np.random.uniform(0.1, 1, size=len(x)), y)

Solution

1. Step 1: Identify correct jitter addition

   Jitter is added by adding small random noise to the x values.

2. Step 2: Check each option

   plt.scatter(x + np.random.uniform(-0.1, 0.1, size=len(x)), y) adds small uniform noise between -0.1 and 0.1, which is correct jitter. Others multiply, subtract large integers, or divide, which distort data incorrectly.

3. Final Answer:

   plt.scatter(x + np.random.uniform(-0.1, 0.1, size=len(x)), y) -> Option A

4. Quick Check:

   Add small noise = C [OK]

Hint: Add small uniform noise to x for jitter [OK]

Common Mistakes:

• Multiplying x by noise instead of adding
• Subtracting large integers causing big shifts
• Dividing x by noise changing scale

3. What will be the effect of the following code on the scatter plot?

import matplotlib.pyplot as plt
import numpy as np

categories = ['A', 'B', 'C']
values = [5, 7, 6]
x = np.array([0, 1, 2])
jitter = np.random.uniform(-0.2, 0.2, size=3)
plt.scatter(x + jitter, values)
plt.xticks(x, categories)
plt.show()

medium

A. Points will be stacked vertically at each category

B. Plot will show an error due to jitter size mismatch

C. Points will be plotted exactly at integer x positions

D. Points will be spread horizontally around each category label

Solution

1. Step 1: Understand jitter addition

   Jitter adds random noise between -0.2 and 0.2 to x positions, spreading points horizontally.

2. Step 2: Effect on scatter plot

   Points will no longer be exactly at 0,1,2 but slightly shifted, making them spread horizontally around category labels.

3. Final Answer:

   Points will be spread horizontally around each category label -> Option D

4. Quick Check:

   Jitter shifts points horizontally = B [OK]

Hint: Jitter shifts points horizontally near categories [OK]

Common Mistakes:

• Thinking jitter stacks points vertically
• Assuming points stay exactly at integers
• Believing code causes error due to jitter size

4. The following code is intended to add jitter to a categorical scatter plot but does not work as expected. What is the main error?

import matplotlib.pyplot as plt
import numpy as np

x = np.array([0, 1, 2, 0, 1, 2])
y = np.array([5, 6, 7, 5, 6, 7])
jitter = np.random.uniform(-0.1, 0.1)
plt.scatter(x + jitter, y)
plt.xticks([0, 1, 2], ['A', 'B', 'C'])
plt.show()

medium

A. Jitter is a single number, not an array matching x length

B. x and y arrays have different lengths

C. plt.xticks labels do not match x values

D. Random uniform range is too large causing overlap

Solution

1. Step 1: Check jitter variable

   Jitter is created as a single float, not an array matching x length.

2. Step 2: Effect on scatter

   Adding a single float to an array works but adds same jitter to all points, no spread.

3. Final Answer:

   Jitter is a single number, not an array matching x length -> Option A

4. Quick Check:

   Jitter must be array for point-wise noise = A [OK]

Hint: Jitter must be array same size as data [OK]

Common Mistakes:

• Assuming single jitter value adds spread
• Ignoring length mismatch of jitter and x
• Thinking xticks labels cause error

5. You have a dataset with categories 'Red', 'Green', 'Blue' and multiple points per category. You want to create a categorical scatter plot with jitter to visualize all points clearly. Which approach is best to implement jitter correctly in matplotlib?

hard

A. Add jitter by multiplying category integers by random values greater than 1

B. Map categories to integers, add small random noise array to these integers, then plot with category labels on x-axis

C. Use bar plot instead of scatter plot to avoid jitter

D. Plot categories as strings on x-axis and add jitter by changing y values randomly

Solution

1. Step 1: Map categories to integers

   Convert categorical labels to integer positions for plotting on x-axis.

2. Step 2: Add small random noise array to integers

   Add jitter by adding a small random noise array to spread points horizontally.

3. Step 3: Plot with category labels

   Use plt.xticks to label integer positions with original category names.

4. Final Answer:

   Map categories to integers, add small random noise array to these integers, then plot with category labels on x-axis -> Option B

5. Quick Check:

   Integer mapping + noise + labels = D [OK]

Hint: Map categories to numbers, add noise, label ticks [OK]

Common Mistakes:

• Adding jitter to y instead of x
• Using bar plot which doesn't show jitter
• Multiplying integers causing large shifts