Scatter plots in Data Analysis Python - Step-by-Step Execution

Concept Flow - Scatter plots

Start with data

↓

Select two variables

↓

Plot points on X and Y axes

↓

Each point shows one data pair

↓

Visualize relationship/trend

↓

Interpret pattern or correlation

Scatter plots show how two variables relate by plotting points on X and Y axes, helping us see patterns or trends.

Execution Sample

Data Analysis Python

import matplotlib.pyplot as plt
x = [1, 2, 3, 4, 5]
y = [2, 3, 5, 7, 11]
plt.scatter(x, y)
plt.show()

This code plots points for x and y values to create a scatter plot showing their relationship.

Execution Table

Step	Action	Variable Values	Plot Points	Output
1	Import matplotlib.pyplot as plt	x=[1,2,3,4,5], y=[2,3,5,7,11]	No points yet	Ready to plot
2	Call plt.scatter(x, y)	x and y unchanged	Points at (1,2), (2,3), (3,5), (4,7), (5,11)	Points placed on plot
3	Call plt.show()	No change	Points visible on graph	Scatter plot displayed
4	End of code	No change	Plot remains visible	Execution stops

💡 All points plotted and plot displayed, code execution ends

Variable Tracker

Variable	Start	After plt.scatter	After plt.show	Final
x	[1, 2, 3, 4, 5]	[1, 2, 3, 4, 5]	[1, 2, 3, 4, 5]	[1, 2, 3, 4, 5]
y	[2, 3, 5, 7, 11]	[2, 3, 5, 7, 11]	[2, 3, 5, 7, 11]	[2, 3, 5, 7, 11]

Key Moments - 3 Insights

Why do the x and y variables not change after plotting?

The execution_table shows x and y keep the same values because plotting only uses them to place points; it does not modify the data.

What does each point on the scatter plot represent?

Each point corresponds to one pair of x and y values, as seen in step 2 where points like (1,2) and (5,11) are plotted.

Why do we call plt.show() after plt.scatter()?

plt.show() displays the plot window with points; without it, the plot may not appear, as shown in step 3.

Visual Quiz - 3 Questions

Look at the execution_table at step 2, what points are plotted?

A (2,1), (3,2), (5,3), (7,4), (11,5)

B (1,2), (2,3), (3,5), (4,7), (5,11)

C (1,1), (2,2), (3,3), (4,4), (5,5)

D No points plotted yet

💡 Hint

Check the 'Plot Points' column in execution_table row for step 2

At which step does the scatter plot become visible?

A Step 1

B Step 2

C Step 3

D Step 4

💡 Hint

Look at the 'Output' column in execution_table for when the plot is displayed

If we change y to [1, 4, 9, 16, 25], how does the plot points row at step 2 change?

A Points become (1,1), (2,4), (3,9), (4,16), (5,25)

B Points remain the same as original y

C Points become (1,2), (2,3), (3,5), (4,7), (5,11)

D No points plotted

💡 Hint

Refer to variable_tracker and execution_table step 2 for how points depend on y values

Concept Snapshot

Scatter plots:
- Plot points for two variables on X and Y axes
- Each point = one data pair
- Use plt.scatter(x, y) in Python
- Call plt.show() to display
- Visualize relationships or trends

Full Transcript

Scatter plots help us see how two sets of numbers relate by drawing points on a graph. We start with two lists of numbers, x and y. Each pair of numbers from x and y forms one point on the plot. Using Python's matplotlib, we call plt.scatter(x, y) to place these points. Then plt.show() displays the plot window so we can see the points. The variables x and y do not change during plotting; they only provide the data. This visual helps us understand if the numbers grow together, have no pattern, or move oppositely.

Practice

1. What does a scatter plot primarily show in data analysis?

easy

A. The relationship between two numerical variables

B. The total count of categories

C. The average value of a dataset

D. The sum of all data points

Solution

1. Step 1: Understand scatter plot purpose

   A scatter plot is used to display how two numerical variables relate to each other by plotting points on a graph.

2. Step 2: Compare options to definition

   Options B, C, and D describe other chart types or calculations, not relationships between two variables.

3. Final Answer:

   The relationship between two numerical variables -> Option A

4. Quick Check:

   Scatter plot = relationship between two numbers [OK]

Hint: Scatter plots always compare two numeric sets [OK]

Common Mistakes:

• Confusing scatter plots with bar charts
• Thinking scatter plots show totals or averages
• Ignoring that scatter plots need two numeric variables

2. Which of the following is the correct way to create a scatter plot using matplotlib in Python?

easy

A. plt.bar(x, y)

B. plt.plot(x, y, kind='scatter')

C. plt.scatterplot(x, y)

D. plt.scatter(x, y)

Solution

1. Step 1: Recall matplotlib scatter syntax

   Matplotlib uses plt.scatter(x, y) to create scatter plots.

2. Step 2: Check other options for correctness

   plt.plot(x, y, kind='scatter') is incorrect syntax; plt.plot does not take kind='scatter'. plt.bar(x, y) creates bar charts. plt.scatterplot(x, y) is not a valid matplotlib function.

3. Final Answer:

   plt.scatter(x, y) -> Option D

4. Quick Check:

   Matplotlib scatter = plt.scatter(x, y) [OK]

Hint: Use plt.scatter(x, y) for scatter plots in matplotlib [OK]

Common Mistakes:

• Using plt.plot with kind='scatter' (wrong syntax)
• Confusing scatter with bar plot functions
• Trying to use plt.scatterplot which doesn't exist

3. What will the following code output?

import matplotlib.pyplot as plt
x = [1, 2, 3, 4]
y = [10, 20, 25, 30]
plt.scatter(x, y)
plt.title('Test Plot')
plt.xlabel('X axis')
plt.ylabel('Y axis')
plt.show()

medium

A. An error because plt.scatter requires more arguments

B. A scatter plot with points at (1,10), (2,20), (3,25), (4,30) and labeled axes

C. A bar chart with bars at x positions 1 to 4

D. A line plot connecting points (1,10), (2,20), (3,25), (4,30)

Solution

1. Step 1: Analyze the code commands

   The code uses plt.scatter(x, y) which plots points at given x and y coordinates. Titles and labels are added correctly.

2. Step 2: Understand the plot output

   The plot will show points at (1,10), (2,20), (3,25), (4,30) with axis labels and a title. No lines or bars are drawn.

3. Final Answer:

   A scatter plot with points at (1,10), (2,20), (3,25), (4,30) and labeled axes -> Option B

4. Quick Check:

   plt.scatter plots points, not lines or bars [OK]

Hint: plt.scatter plots points, not lines or bars [OK]

Common Mistakes:

• Thinking plt.scatter draws lines between points
• Confusing scatter plot with bar chart
• Assuming plt.scatter needs extra arguments

4. Identify the error in this code snippet that tries to create a scatter plot:

import matplotlib.pyplot as plt
x = [1, 2, 3]
y = [4, 5]
plt.scatter(x, y)
plt.show()

medium

A. plt.scatter requires a third argument for color

B. plt.show() is missing parentheses

C. x and y lists have different lengths causing an error

D. No error, code runs fine

Solution

1. Step 1: Check lengths of x and y lists

   x has 3 elements, y has 2 elements. Scatter plot requires equal length for x and y.

2. Step 2: Understand matplotlib error behavior

   Matplotlib will raise a ValueError because x and y lengths differ. Options A and C are incorrect because color is optional and plt.show() is correctly called.

3. Final Answer:

   x and y lists have different lengths causing an error -> Option C

4. Quick Check:

   Equal length x and y needed for scatter [OK]

Hint: x and y must be same length for scatter plots [OK]

Common Mistakes:

• Ignoring length mismatch between x and y
• Thinking color argument is mandatory
• Assuming plt.show() can be called without parentheses

5. You have a dataset with columns 'Age' and 'Income'. You want to create a scatter plot to find if older people tend to earn more. Which steps should you follow to create a clear scatter plot in Python using pandas and matplotlib?

hard

A. Import pandas and matplotlib, extract 'Age' and 'Income' columns as x and y, use plt.scatter(x, y), add title and axis labels, then plt.show()

B. Use plt.bar with 'Age' as x and 'Income' as height, then add labels

C. Plot a line graph with 'Age' on x-axis and 'Income' on y-axis without labels

D. Create a pie chart showing distribution of 'Income' grouped by 'Age' categories

Solution

1. Step 1: Extract relevant columns for scatter plot

   Use pandas to get 'Age' and 'Income' columns as x and y arrays for plotting.

2. Step 2: Plot scatter and add labels

   Use plt.scatter(x, y) to plot points, add a title and axis labels for clarity, then display with plt.show().

3. Final Answer:

   Import pandas and matplotlib, extract 'Age' and 'Income' columns as x and y, use plt.scatter(x, y), add title and axis labels, then plt.show() -> Option A

4. Quick Check:

   Scatter plot with labels = Import pandas and matplotlib, extract 'Age' and 'Income' columns as x and y, use plt.scatter(x, y), add title and axis labels, then plt.show() [OK]

Hint: Extract columns, use plt.scatter, add labels, then show plot [OK]

Common Mistakes:

• Using bar or pie charts instead of scatter for relationships
• Skipping axis labels and titles
• Plotting line graph which implies continuous trend