Reader small image

You're reading from  Jupyter Cookbook

Product typeBook
Published inApr 2018
Reading LevelIntermediate
PublisherPackt
ISBN-139781788839440
Edition1st Edition
Languages
Tools
Right arrow
Author (1)
Dan Toomey
Dan Toomey
author image
Dan Toomey

Dan Toomey has been developing application software for over 20 years. He has worked in a variety of industries and companies, in roles from sole contributor to VP/CTO-level. For the last few years, he has been contracting for companies in the eastern Massachusetts area. Dan has been contracting under Dan Toomey Software Corp. Dan has also written R for Data Science, Jupyter for Data Sciences, and the Jupyter Cookbook, all with Packt.
Read more about Dan Toomey

Right arrow

Generate a regression line of data using R


In this example, we use the abline function to portray a regression line of our data.

How to do it...

We can use this script:

# load the iris dataset
data <- read.csv("http://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data")

#Let us also clean up the data so as to be more readable
colnames(data) <- c("sepal_length", "sepal_width", "petal_length", "petal_width", "species")

# call plot first
plot(data$sepal_length, data$petal_length)

# abline adds to the plot
abline(lm(data$petal_length ~ sepal_length), col="red")

It results in a similar Scatter plot but with a regression line included:

How it works...

We are using the same iris dataset as in the previous example.

We have seen how plot can produce a Scatter plot. The addition by abline is to calculate and draw out the regression line on top of the Scatter plot.

The regression does not appear to be a great fit as there are big chunks of data points far away from the line.

lock icon
The rest of the page is locked
Previous PageNext Page
You have been reading a chapter from
Jupyter Cookbook
Published in: Apr 2018Publisher: PacktISBN-13: 9781788839440

Author (1)

author image
Dan Toomey

Dan Toomey has been developing application software for over 20 years. He has worked in a variety of industries and companies, in roles from sole contributor to VP/CTO-level. For the last few years, he has been contracting for companies in the eastern Massachusetts area. Dan has been contracting under Dan Toomey Software Corp. Dan has also written R for Data Science, Jupyter for Data Sciences, and the Jupyter Cookbook, all with Packt.
Read more about Dan Toomey