Linear regression
We will assume that fertility is a linear function of the female illiteracy rate. That is, \(f = a i + b\), where \(a\) is the slope and \(b\) is the intercept. We can think of the intercept as the minimal fertility rate, probably somewhere between one and two. The slope tells us how the fertility rate varies with illiteracy. We can find the best fit line using np.polyfit()
.
Plot the data and the best fit line. Print out the slope and intercept. (Think: what are their units?)
This exercise is part of the course
Statistical Thinking in Python (Part 2)
Exercise instructions
- Compute the slope and intercept of the regression line using
np.polyfit()
. Remember,fertility
is on the y-axis andilliteracy
on the x-axis. - Print out the slope and intercept from the linear regression.
- To plot the best fit line, create an array
x
that consists of 0 and 100 usingnp.array()
. Then, compute the theoretical values ofy
based on your regression parameters. I.e.,y = a * x + b
. - Plot the data and the regression line on the same plot. Be sure to label your axes.
- Hit submit to display your plot.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Plot the illiteracy rate versus fertility
_ = plt.plot(illiteracy, fertility, marker='.', linestyle='none')
plt.margins(0.02)
_ = plt.xlabel('percent illiterate')
_ = plt.ylabel('fertility')
# Perform a linear regression using np.polyfit(): a, b
a, b = ____
# Print the results to the screen
print('slope =', a, 'children per woman / percent illiterate')
print('intercept =', b, 'children per woman')
# Make theoretical line to plot
x = ____
y = ____ * ____ + ____
# Add regression line to your plot
_ = plt.plot(____, ____)
# Draw the plot
plt.show()