Trend lines on scatter plots in Matplotlib - Cheat Sheet & Quick Revision

Recall & Review

beginner

What is a trend line in a scatter plot?

A trend line is a straight or curved line drawn through a scatter plot to show the general direction or pattern of the data points.

beginner

How do you calculate a simple linear trend line for scatter plot data?

You calculate the best-fit line using linear regression, which finds the line that minimizes the sum of the squared distances between itself and all data points.

beginner

Which Python library is commonly used to add trend lines to scatter plots?

Matplotlib is commonly used for plotting, and NumPy or SciPy can be used to calculate trend lines with linear regression.

beginner

What does the slope of a trend line represent?

The slope shows how much the y-value changes for each unit increase in the x-value, indicating the strength and direction of the relationship.

beginner

Why is it useful to add a trend line to a scatter plot?

A trend line helps to see the overall pattern or relationship in noisy data, making it easier to understand and explain.

What function in NumPy can be used to calculate the coefficients for a linear trend line?

A numpy.polyfit()

B numpy.mean()

C numpy.random()

D numpy.sort()

numpy.polyfit() fits a polynomial (degree 1 for linear) to data points, giving coefficients for the trend line.

In matplotlib, which function is used to create a scatter plot?

A plt.plot()

B plt.scatter()

C plt.bar()

D plt.hist()

plt.scatter() creates scatter plots by plotting individual points.

What does a positive slope in a trend line indicate?

A As x increases, y decreases

B No relationship between x and y

C As x increases, y increases

D Data points are random

A positive slope means y values go up as x values go up.

Which of these is NOT a reason to add a trend line to a scatter plot?

A To make the plot colorful

B To predict future values

C To highlight the general pattern

D To summarize data relationship

Trend lines are for analysis, not for decoration or color.

What degree polynomial fits a straight line trend?

A 3

B 0

C 2

D 1

Degree 1 polynomial is a straight line.

Explain how to add a linear trend line to a scatter plot using matplotlib and numpy.

Think about fitting a line and then drawing it over the scatter points.

Describe why trend lines are helpful when analyzing scatter plot data.

Consider what you want to learn from many scattered points.

Practice

1. What is the main purpose of adding a trend line to a scatter plot in matplotlib?

easy

A. To show the overall pattern or relationship between variables

B. To add colors to the scatter points

C. To change the size of scatter points

D. To remove outliers from the data

Solution

1. Step 1: Understand scatter plot purpose

   A scatter plot shows individual data points but may not clearly show the overall trend.

2. Step 2: Role of trend line

   A trend line summarizes the general direction or pattern in the data, making relationships easier to see.

3. Final Answer:

   To show the overall pattern or relationship between variables -> Option A

4. Quick Check:

   Trend line = overall pattern [OK]

Hint: Trend lines summarize data direction, not point details [OK]

Common Mistakes:

• Thinking trend lines change point colors
• Confusing trend lines with point size changes
• Believing trend lines remove data points

2. Which of the following code snippets correctly fits a linear trend line to data arrays x and y using numpy?

easy

A. coefficients = np.polyfit(x, y, 1)

B. coefficients = np.polyfit(y, x, 1)

C. coefficients = np.polyfit(x, y, 2)

D. coefficients = np.polyfit(x, y)

Solution

1. Step 1: Understand np.polyfit parameters

   np.polyfit(x, y, degree) fits a polynomial of given degree to x and y data.

2. Step 2: Correct usage for linear fit

   For a linear trend line, degree = 1, and x is independent variable, y is dependent variable.

3. Final Answer:

   coefficients = np.polyfit(x, y, 1) -> Option A

4. Quick Check:

   np.polyfit(x, y, 1) fits linear trend [OK]

Hint: Use degree=1 for linear trend with np.polyfit(x, y, 1) [OK]

Common Mistakes:

• Swapping x and y in np.polyfit
• Omitting degree parameter
• Using degree 2 for linear trend

3. Given the code below, what will be the output of print(trendline(3))?

import numpy as np
x = np.array([1, 2, 3, 4])
y = np.array([2, 4, 6, 8])
coefficients = np.polyfit(x, y, 1)
trendline = np.poly1d(coefficients)
print(trendline(3))

medium

A. Error due to incorrect polyfit usage

B. 5.0

C. 9.0

D. 6.0

Solution

1. Step 1: Analyze data and polyfit

   Data y doubles x (2,4,6,8), so linear fit slope ~2 and intercept ~0.

2. Step 2: Calculate trendline at x=3

   trendline(3) = 2*3 + 0 = 6.0

3. Final Answer:

   6.0 -> Option D

4. Quick Check:

   trendline(3) = 6.0 [OK]

Hint: Check if y is roughly 2*x; then trendline(3) = 6 [OK]

Common Mistakes:

• Confusing x and y order in polyfit
• Forgetting poly1d creates function
• Assuming error instead of numeric output

4. Identify the error in the following code that tries to plot a scatter plot with a trend line:

import numpy as np
import matplotlib.pyplot as plt
x = np.array([1, 2, 3, 4])
y = np.array([2, 3, 5, 7])
coeff = np.polyfit(x, y, 1)
trend = np.poly1d(coeff)
plt.scatter(x, y)
plt.plot(x, trend)
plt.show()

medium

A. np.polyfit arguments are in wrong order

B. plt.plot(x, trend) should be plt.plot(x, trend(x))

C. plt.scatter should be called after plt.plot

D. Missing plt.xlabel and plt.ylabel calls

Solution

1. Step 1: Check trend line plotting

   plt.plot expects y-values, but trend is a function, not values.

2. Step 2: Correct function call

   Call trend(x) to get y-values for plotting: plt.plot(x, trend(x))

3. Final Answer:

   plt.plot(x, trend) should be plt.plot(x, trend(x)) -> Option B

4. Quick Check:

   Plot function needs y-values, not function [OK]

Hint: Call trend(x) to get y-values for plt.plot [OK]

Common Mistakes:

• Passing function instead of values to plt.plot
• Swapping x and y in polyfit
• Thinking plot order affects output
• Missing axis labels is not an error

5. You have noisy data with outliers. Which approach best improves the trend line fit on a scatter plot using matplotlib and numpy?

hard

A. Plot scatter without trend line to avoid confusion

B. Use np.polyfit with degree 1 on all data points

C. Remove outliers before fitting trend line with np.polyfit

D. Use plt.plot instead of plt.scatter for all points

Solution

1. Step 1: Understand effect of outliers

   Outliers can distort the trend line fit, making it less representative.

2. Step 2: Improve fit by removing outliers

   Removing or filtering outliers before fitting helps get a better trend line.

3. Final Answer:

   Remove outliers before fitting trend line with np.polyfit -> Option C

4. Quick Check:

   Remove outliers for better trend line [OK]

Hint: Filter outliers before fitting trend line for accuracy [OK]

Common Mistakes:

• Fitting trend line on all data including outliers
• Skipping trend line when data is noisy
• Confusing plot types for trend line fitting