Before-after comparison plots in Matplotlib

Introduction

Before-after comparison plots help us see changes clearly by showing data from two times side by side.

To compare test scores of students before and after a training program.

To check sales numbers before and after a marketing campaign.

To observe weight changes before and after a diet plan.

To analyze temperature differences before and after installing new equipment.

Syntax

Matplotlib

import matplotlib.pyplot as plt

# Data for before and after
before = [value1, value2, ...]
after = [value1, value2, ...]

# Create x positions
x = range(len(before))

# Plot lines connecting before and after
plt.plot(x, before, 'o-', label='Before')
plt.plot(x, after, 's-', label='After')

# Add labels and legend
plt.xlabel('Items')
plt.ylabel('Values')
plt.title('Before-After Comparison')
plt.legend()
plt.show()

Use markers like 'o' and 's' to show points clearly.

Connecting lines help visualize changes between before and after.

Examples

Simple plot with default x positions (0,1,2) showing before and after values.

Matplotlib

before = [5, 7, 9]
after = [6, 8, 10]

plt.plot(before, 'o-', label='Before')
plt.plot(after, 's-', label='After')
plt.legend()
plt.show()

Using custom x labels to represent categories.

Matplotlib

x = ['A', 'B', 'C']
before = [3, 4, 5]
after = [4, 5, 6]

plt.plot(x, before, 'o-', label='Before')
plt.plot(x, after, 's-', label='After')
plt.legend()
plt.show()

Adding dashed lines to connect before and after points for each item.

Matplotlib

before = [10, 15, 20]
after = [12, 18, 22]
x = range(len(before))

for i in x:
    plt.plot([i, i], [before[i], after[i]], 'k--')
plt.plot(x, before, 'o-', label='Before')
plt.plot(x, after, 's-', label='After')
plt.legend()
plt.show()

Sample Program

This code shows sales data before and after a campaign for 5 months. Dashed lines connect each month's before and after sales to highlight changes.

Matplotlib

import matplotlib.pyplot as plt

# Sample data: sales before and after a campaign
before = [100, 150, 200, 250, 300]
after = [120, 160, 210, 280, 330]

x = range(len(before))

# Plot lines connecting before and after values
for i in x:
    plt.plot([i, i], [before[i], after[i]], 'gray', linestyle='--')

plt.plot(x, before, 'o-', color='blue', label='Before')
plt.plot(x, after, 's-', color='green', label='After')

plt.xlabel('Month')
plt.ylabel('Sales')
plt.title('Sales Before and After Campaign')
plt.legend()
plt.show()

Important Notes

Make sure before and after lists have the same length.

Use clear labels and legends to help understand the plot.

Colors and markers make it easier to distinguish before and after data.

Summary

Before-after plots show changes clearly by plotting two sets of data points.

Connecting lines help visualize the difference for each item.

Use labels, legends, and colors to make the plot easy to read.

Practice

1. What is the main purpose of a before-after comparison plot in matplotlib?

easy

A. To visually compare data from two different time points

B. To show the distribution of a single dataset

C. To display the correlation between two variables

D. To create a 3D surface plot

Solution

1. Step 1: Understand the concept of before-after plots

   Before-after plots are used to compare data points from two different times or conditions to see changes.

2. Step 2: Identify the correct purpose

   Among the options, only To visually compare data from two different time points describes comparing data from two time points, which matches the before-after plot purpose.

3. Final Answer:

   To visually compare data from two different time points -> Option A

4. Quick Check:

   Before-after plots = compare two time points [OK]

Hint: Before-after plots compare two sets of data visually [OK]

Common Mistakes:

• Confusing before-after plots with distribution plots
• Thinking they show correlation instead of change
• Assuming they create 3D plots

2. Which of the following is the correct way to plot two sets of data side-by-side for before-after comparison using matplotlib?

easy

A. plt.plot(before_data); plt.plot(after_data)

B. plt.bar([0,1], before_data); plt.bar([0,1], after_data)

C. plt.bar([0,1], before_data); plt.bar([1,2], after_data)

D. plt.bar([1,2], [before_data, after_data])

Solution

1. Step 1: Understand bar plot positioning

   To show before and after side-by-side, bars must not overlap. Using different x positions for before and after data avoids overlap.

2. Step 2: Analyze options for correct bar positions

   plt.bar([0,1], before_data); plt.bar([1,2], after_data) places before_data at positions 0 and 1, and after_data at 1 and 2, so bars for the same category are side-by-side without overlap.

3. Final Answer:

   plt.bar([0,1], before_data); plt.bar([1,2], after_data) -> Option C

4. Quick Check:

   Side-by-side bars need different x positions [OK]

Hint: Use different x positions to avoid bar overlap [OK]

Common Mistakes:

• Plotting bars at same x positions causing overlap
• Using plt.plot instead of plt.bar for categorical data
• Passing data incorrectly as list of lists

3. What will be the output of this code snippet?

import matplotlib.pyplot as plt
before = [5, 7]
after = [8, 6]
plt.plot([1, 2], before, label='Before')
plt.plot([1, 2], after, label='After')
plt.legend()
plt.show()

medium

A. An error because plt.plot cannot take two lists

B. A bar chart comparing before and after data

C. A scatter plot with points at (1,5), (2,7), (1,8), (2,6)

D. Two overlapping line plots showing before and after data

Solution

1. Step 1: Understand plt.plot with x and y lists

   plt.plot([1, 2], before) plots points (1,5) and (2,7) connected by a line. Similarly for after data.

2. Step 2: Identify plot type and legend

   Two line plots will appear overlapping on the same axes with labels 'Before' and 'After'. No error occurs.

3. Final Answer:

   Two overlapping line plots showing before and after data -> Option D

4. Quick Check:

   plt.plot with x,y lists = line plot [OK]

Hint: plt.plot(x, y) draws lines connecting points [OK]

Common Mistakes:

• Thinking plt.plot creates bar charts
• Expecting scatter plot without plt.scatter
• Assuming plt.plot with two lists causes error

4. Identify the error in this code for before-after bar plot:

import matplotlib.pyplot as plt
before = [3, 4]
after = [5, 6]
plt.bar([0, 1], before)
plt.bar([0, 1], after)
plt.show()

medium

A. plt.show() is missing

B. Bars for before and after overlap at same positions

C. before and after lists must be same length

D. plt.bar requires three arguments

Solution

1. Step 1: Check bar positions

   Both before and after bars are plotted at positions 0 and 1, causing them to overlap and hide one another.

2. Step 2: Identify correct fix

   To avoid overlap, after bars should be shifted to different x positions, e.g., [0.3, 1.3].

3. Final Answer:

   Bars for before and after overlap at same positions -> Option B

4. Quick Check:

   Same x positions cause bar overlap [OK]

Hint: Shift bars on x-axis to avoid overlap [OK]

Common Mistakes:

• Thinking plt.bar needs 3 arguments
• Ignoring bar overlap issue
• Assuming plt.show() is missing

5. You have sales data before and after a marketing campaign for 3 products: before = [100, 150, 200], after = [120, 180, 210]. How would you create a clear before-after bar plot with labels and legend in matplotlib?

hard

A. Use plt.bar with shifted x positions for before and after, add labels and legend

B. Plot before and after using plt.plot without labels

C. Use plt.scatter for both datasets on same x positions

D. Plot only after data as a bar chart

Solution

1. Step 1: Plan bar positions and labels

   To compare before and after clearly, plot bars side-by-side with shifted x positions, e.g., before at [0,1,2], after at [0.3,1.3,2.3]. Add x-axis labels for products.

2. Step 2: Add legend and labels for clarity

   Use plt.legend() to distinguish before and after bars, and plt.xlabel/plt.ylabel for axis labels.

3. Final Answer:

   Use plt.bar with shifted x positions for before and after, add labels and legend -> Option A

4. Quick Check:

   Shift bars + labels + legend = clear before-after plot [OK]

Hint: Shift bars and add legend for clear comparison [OK]

Common Mistakes:

• Plotting bars at same positions causing confusion
• Skipping labels and legend
• Using scatter plot instead of bar plot