Javascript required
Skip to content Skip to sidebar Skip to footer

Draw Regression Line on Plot

In order to focus on the construction of the different plots and the use of {ggplot2}, we will restrict ourselves to drawing basic (yet beautiful) plots without unnecessary layers. For the sake of completeness, we will briefly discuss and illustrate different layers to further personalize a plot at the end of the article (see this section).

Note that if you still struggle to create plots with {ggplot2} after reading this tutorial, you may find the {esquisse} addin useful. This addin allows you to interactively (that is, by dragging and dropping variables) create plots with the {ggplot2} package. Give it a try!

Scatter plot

We start by creating a scatter plot using geom_point. Remember that a scatter plot is used to visualize the relation between two quantitative variables.

  1. We start by specifying the data:
          ggplot(dat) # data        

  1. Then we add the variables to be represented with the aes() function:
          ggplot(dat) + # data   aes(x = displ, y = hwy) # variables        

  1. Finally, we indicate the type of plot:
          ggplot(dat) + # data   aes(x = displ, y = hwy) + # variables   geom_point() # type of plot        

You will also sometimes see the aesthetic elements (aes() with the variables) inside the ggplot() function in addition to the dataset:

          ggplot(mpg, aes(x = displ, y = hwy)) +   geom_point()        

This second method gives the exact same plot than the first method. I tend to prefer the first method over the second for better readability, but this is more a matter of taste so the choice is up to you.

Line plot

Line plots, particularly useful in time series or finance, can be created similarly but by using geom_line():

          ggplot(dat) +   aes(x = displ, y = hwy) +   geom_line()        

Combination of line and points

An advantage of {ggplot2} is the ability to combine several types of plots and its flexibility in designing it. For instance, we can add a line to a scatter plot by simply adding a layer to the initial scatter plot:

          ggplot(dat) +   aes(x = displ, y = hwy) +   geom_point() +   geom_line() # add line        

Histogram

A histogram (useful to visualize distributions and detect potential outliers) can be plotted using geom_histogram():

          ggplot(dat) +   aes(x = hwy) +   geom_histogram()        

By default, the number of bins is equal to 30. You can change this value using the bins argument inside the geom_histogram() function:

          ggplot(dat) +   aes(x = hwy) +   geom_histogram(bins = sqrt(nrow(dat)))        

Here I specify the number of bins to be equal to the square root of the number of observations (following Sturge's rule) but you can specify any numeric value.

Density plot

Density plots can be created using geom_density():

          ggplot(dat) +   aes(x = hwy) +   geom_density()        

Combination of histogram and densities

We can also superimpose a histogram and a density curve on the same plot:

          ggplot(dat) +   aes(x = hwy, y = ..density..) +   geom_histogram() +   geom_density()        

Or superimpose several densities:

          ggplot(dat) +   aes(x = hwy, color = drv, fill = drv) +   geom_density(alpha = 0.25) # add transparency        

The argument alpha = 0.25 has been added for some transparency. More information about this argument can be found in this section.

Boxplot

A boxplot (also very useful to visualize distributions and detect potential outliers) can be plotted using geom_boxplot():

          # Boxplot for one variable ggplot(dat) +   aes(x = "", y = hwy) +   geom_boxplot()        

          # Boxplot by factor ggplot(dat) +   aes(x = drv, y = hwy) +   geom_boxplot()        

It is also possible to plot the points on the boxplot with geom_jitter(), and to vary the width of the boxes according to the size (i.e., the number of observations) of each level with varwidth = TRUE:

          ggplot(dat) +   aes(x = drv, y = hwy) +   geom_boxplot(varwidth = TRUE) + # vary boxes width according to n obs.   geom_jitter(alpha = 0.25, width = 0.2) # adds random noise and limit its width        

The geom_jitter() layer adds some random variation to each point in order to prevent them from overlapping (an issue known as overplotting).1 Moreover, the alpha argument adds some transparency to the points (see more in this section) to keep the focus on the boxes and not on the points.

Finally, it is also possible to divide boxplots into several panels according to the levels of a qualitative variable:

          ggplot(dat) +   aes(x = drv, y = hwy) +   geom_boxplot(varwidth = TRUE) + # vary boxes width according to n obs.   geom_jitter(alpha = 0.25, width = 0.2) + # adds random noise and limit its width   facet_wrap(~year) # divide into 2 panels        

For a visually more appealing plot, it is also possible to use some colors for the boxes depending on the x variable:

          ggplot(dat) +   aes(x = drv, y = hwy, fill = drv) + # add color to boxes with fill   geom_boxplot(varwidth = TRUE) + # vary boxes width according to n obs.   geom_jitter(alpha = 0.25, width = 0.2) + # adds random noise and limit its width   facet_wrap(~year) + # divide into 2 panels   theme(legend.position = "none") # remove legend        

In that case, it best to remove the legend as it becomes redundant. See more information about the legend in this section.

If you are unhappy with the default colors provided in {ggplot2}, you can change them manually with the scale_fill_manual() layer:

          ggplot(dat) +   aes(x = drv, y = hwy, fill = drv) + # add color to boxes with fill   geom_boxplot(varwidth = TRUE) + # vary boxes width according to n obs.   geom_jitter(alpha = 0.25, width = 0.2) + # adds random noise and limit its width   facet_wrap(~year) + # divide into 2 panels   theme(legend.position = "none") + # remove legend   scale_fill_manual(values = c("darkred", "darkgreen", "steelblue")) # change fill color manually        

Barplot

A barplot (useful to visualize qualitative variables) can be plotted using geom_bar():

          ggplot(dat) +   aes(x = drv) +   geom_bar()        

By default, the heights of the bars correspond to the observed frequencies for each level of the variable of interest (drv in our case).

Again, for a more appealing plot, we can add some colors to the bars with the fill argument:

          ggplot(dat) +   aes(x = drv, fill = drv) + # add colors to bars   geom_bar() +   theme(legend.position = "none") # remove legend        

We can also create a barplot with two qualitative variables:

          ggplot(dat) +   aes(x = drv, fill = year) + # fill by years   geom_bar()        

In order to compare proportions across groups, it is best to make each bar the same height using position = "fill":

          ggplot(dat) +   geom_bar(aes(x = drv, fill = year), position = "fill")        

To draw the bars next to each other for each group, use position = "dodge":

          ggplot(dat) +   geom_bar(aes(x = drv, fill = year), position = "dodge")        

Further personalization

Title and axis labels

The first things to personalize in a plot is the labels to make the plot more informative to the audience. We can easily add a title, subtitle, caption and edit axis labels with the labs() function:

            p <- ggplot(dat) +   aes(x = displ, y = hwy) +   geom_point()  p + labs(   title = "Fuel efficiency for 38 popular models of car",   subtitle = "Period 1999-2008",   caption = "Data: ggplot2::mpg. See more at statsandr.com",   x = "Engine displacement (litres)",   y = "Highway miles per gallon (mpg)" )          

As you can see in the above code, you can save one or more layers of the plot in an object for later use. This way, you can save your "main" plot, and add more layers of personalization until you get the desired output. Here we saved the main scatter plot in an object called p and we will refer to it for the subsequent personalizations.

You can also edit the alignment, the size and the shape of the title and subtitle via the theme() layer and the element_text() function:

            p + labs(   title = "Fuel efficiency for 38 popular models of car",   subtitle = "Period 1999-2008",   caption = "Data: ggplot2::mpg. See more at statsandr.com",   x = "Engine displacement (litres)",   y = "Highway miles per gallon (mpg)" ) +   theme(     plot.title = element_text(       hjust = 0.5, # center       size = 12,       color = "steelblue",       face = "bold"     ),     plot.subtitle = element_text(       hjust = 0.5, # center       size = 10,       color = "gray",       face = "italic"     )   )          

If the title or subtitle is long and you want to divide it into multiple lines, use \n:

            p + labs(   title = "Fuel efficiency for 38 popular \n models of car",   subtitle = "Period 1999-2008",   caption = "Data: ggplot2::mpg. See more at statsandr.com",   x = "Engine displacement (litres)",   y = "Highway miles per gallon (mpg)" ) +   theme(     plot.title = element_text(       hjust = 0.5, # center       size = 12,       color = "steelblue",       face = "bold"     ),     plot.subtitle = element_text(       hjust = 0.5, # center       size = 10,       color = "gray",       face = "italic"     )   )          

Axis ticks

Axis ticks can be adjusted using scale_x_continuous() and scale_y_continuous() for the x and y-axis, respectively:

            # Adjust ticks p + scale_x_continuous(breaks = seq(from = 1, to = 7, by = 0.5)) + # x-axis   scale_y_continuous(breaks = seq(from = 10, to = 45, by = 5)) # y-axis          

Log transformations

In some cases, it is useful to plot the log transformation of the variables. This can be done with the scale_x_log10() and scale_y_log10() functions:

            p + scale_x_log10() +   scale_y_log10()          

Limits

The most convenient way to control the limits of the plot is to use again the scale_x_continuous() and scale_y_continuous() functions in addition to the limits argument:

            p + scale_x_continuous(limits = c(3, 6)) +   scale_y_continuous(limits = c(20, 30))          

It is also possible to simply take a subset of the dataset with the subset() or filter() function. See how to subset a dataset if you need a reminder.

Scales for better axis formats

Depending on your data, it is possible to format axes in a certain way with the {scales} package. The formats I use the most are comma and label_number_si() which format large numbers in a more-readable way.

For this example, we multiply both variables by 1000 to have larger numbers and then we apply a different format to each axis:

            ggplot(dat) +   aes(x = displ * 1000, y = hwy * 1000) +   geom_point() +   scale_y_continuous(labels = scales::label_number_si()) + # format y-axis   scale_x_continuous(labels = scales::comma) # format x-axis          

As you can see, numbers on the y-axis are automatically labeled with the best SI prefix ("K" for values \(\ge\) 10e3, "M" for \(\ge\) 10e6, "B" for \(\ge\) 10e9, and "T" for \(\ge\) 10e12) and numbers on the x-axis are displayed as 2,000, 3,000, etc. instead of 2000, 3000, etc.

These two formats make large numbers easier to read. Other formats are possible such as using dollar signs, percentage signs, dates etc. See more information in the package's documentation.

Legend

By default, the legend is located to the right side of the plot (when there is a legend to be displayed of course). To control the position of the legend, we need to use the theme() function in addition to the legend.position argument:

            p + aes(color = class) +   theme(legend.position = "top")          

Replace "top" by "left" or "bottom" to change its position and by "none" to remove it.

The title of the legend can be edited with the labs() layer:

            p + aes(color = class) +   labs(color = "Car's class")          

Note that the argument inside labs() must match the one inside the aes() layer (in this case: color).

The title of the legend can also be removed with legend.title = element_blank() inside the theme() layer:

            p + aes(color = class) +   theme(     legend.title = element_blank(),     legend.position = "bottom"   )          

The legend now appears at the bottom of the plot, without the legend title.

Shape, color, size and transparency

There are a very large number of options to improve the quality of the plot or to add additional information. These include:

  • shape,
  • size,
  • color, and
  • alpha (transparency).

We can for instance change the shape of all points in a scatter plot by adding shape to geom_point(), or vary the shape according to the values taken by another variable (in that case, the shape argument must be inside aes()):2

            # Change shape of all points ggplot(dat) +   aes(x = displ, y = hwy) +   geom_point(shape = 4)          

            # Change shape of points based on a categorical variable ggplot(dat) +   aes(x = displ, y = hwy, shape = drv) +   geom_point()          

Following the same principle, we can modify the color, size and transparency of the points based on a qualitative or quantitative variable. Here are some examples:

            p <- ggplot(dat) +   aes(x = displ, y = hwy) +   geom_point()  # Change color for all points p + geom_point(color = "steelblue")          

            # Change color based on a qualitative variable p + aes(color = drv)          

            # Change color based on a quantitative variable p + aes(color = cty)          

            # Change color based on a criterion (median of cty variable) p + aes(color = cty > median(cty))          

            # Change size of all points p + geom_point(size = 4)          

            # Change size of points based on a quantitative variable p + aes(size = cty)          

            # Change transparency based on a quantitative variable p + aes(alpha = cty)          

We can of course mix several options (shape, color, size, alpha) to build more complex graphics:

            p + geom_point(size = 0.5) +   aes(color = drv, shape = year, alpha = cty)          

If you are unhappy with the default colors, you can change them manually with the scale_colour_manual() layer (for qualitative variables) and the scale_coulour_gradient2() layer (for quantitative variables):

            # Change color based on a qualitative variable p + aes(color = drv) +   scale_colour_manual(values = c("red", "blue", "green"))          

            # Change color based on a quantitative variable p + aes(color = cty) +   scale_colour_gradient2(     low = "green",     mid = "gray",     high = "red",     midpoint = median(dat$cty)   )          

For your information, you can emulate {ggplot2} default color palette for a desired number of colors and produce a character vector of HEX colors. For example, with 4 colors:

            library(scales) show_col(hue_pal()(4))          

Text and labels

To add a label on a point (for example the row number), we can use the geom_text() and aes() functions:

            p + geom_text(aes(label = rownames(dat)),   check_overlap = TRUE,   size = 2,   vjust = -1 )          

To add text on the plot, we use the annotate() function:

            p + annotate("text",   x = 6,   y = 40,   label = "hwy and displ are \n negatively correlated \n (rho = -0.77, p-value < 0.001)",   size = 3 )          

Read the article on correlation coefficient and correlation test in R to see how I computed the correlation coefficient (rho) and the p-value of the correlation test.

Smooth and regression lines

In a scatter plot, it is possible to add a smooth line fitted to the data:

            p + geom_smooth()          

In the context of simple linear regression, it is often the case that the regression line is displayed on the plot. This can be done by adding method = lm (lm stands for linear model) in the geom_smooth() layer:

            p + geom_smooth(method = lm)          

It is also possible to draw a regression line for each level of a categorical variable:

            p + aes(color = drv, shape = drv) +   geom_smooth(method = lm, se = FALSE)          

The se = FALSE argument removes the confidence interval around the regression lines.

Facets

facet_grid allows you to divide the same graphic into several panels according to the values of one or two qualitative variables:

            # According to one variable p + facet_grid(. ~ drv)          

            # According to 2 variables p + facet_grid(drv ~ year)          

It is then possible to add a regression line to each facet:

            p + facet_grid(. ~ drv) +   geom_smooth(method = lm)          

facet_wrap() can also be used, as illustrated in this section.

Themes

Several functions are available in the {ggplot2} package to change the theme of the plot. The most common themes after the default theme (i.e., theme_gray()) are the black and white (theme_bw()), minimal (theme_minimal()) and classic (theme_classic()) themes:

            # Black and white theme p + theme_bw()          

            # Minimal theme p + theme_minimal()          

            # Classic theme p + theme_classic()          

I tend to use the minimal theme for most of my R Markdown reports as it brings out the patterns and points and not the layout of the plot, but again this is a matter of personal taste. See more themes at ggplot2.tidyverse.org/reference/ggtheme.html and in the {ggthemes} package.

In order to avoid having to change the theme for each plot you create, you can change the theme for the current R session using the theme_set() function as follows:

            theme_set(theme_minimal())          

Interactive plot with {plotly}

You can easily make your plots created with {ggplot2} interactive with the {plotly} package:

            library(plotly) ggplotly(p + aes(color = year))          

You can now hover over a point to display more information about that point. There is also the possibility to zoom in and out, to download the plot, to select some observations, etc. More information about {plotly} for R can be found here.

Combine plots with {patchwork}

There are several ways to combine plots made in {ggplot2}. In my opinion, the most convenient way is with the {patchwork} package using symbols such as +, / and parentheses.

We first need to create some plots and save them:

            p_a <- ggplot(dat) +   aes(x = displ, y = hwy) +   geom_point()  p_b <- ggplot(dat) +   aes(x = hwy) +   geom_histogram()  p_c <- ggplot(dat) +   aes(x = drv, y = hwy) +   geom_boxplot()          

Now that we have 3 plots saved in our environment, we can combine them. To have plots next to each other simply use the + symbol:

            library(patchwork) p_a + p_b + p_c          

To display them above each other simply use the / symbol:

            p_a / p_b / p_c          

And finally, to combine them above and next to each other, mix +, / and parentheses:

            p_a + p_b / p_c          

            (p_a + p_b) / p_c          

See more ways to combine plots with:

  • grid.arrange() from the {gridExtra} package
  • plot_grid() from the {cowplot} package

Flip coordinates

Flipping coordinates of your plot is useful to create horizontal boxplots, or when labels of a variable are so long that they overlap each other on the x-axis. See with and without flipping coordinates below:

            # without flipping coordinates p1 <- ggplot(dat) +   aes(x = class, y = hwy) +   geom_boxplot()  # with flipping coordinates p2 <- ggplot(dat) +   aes(x = class, y = hwy) +   geom_boxplot() +   coord_flip()  library(patchwork) p1 + p2 # left: without flipping, right: with flipping          

This can be done with many types of plot, not only with boxplots. For instance, if a categorical variable has many levels or the labels are long, it is usually best to flip the coordinates for a better visual:

            ggplot(dat) +   aes(x = class) +   geom_bar() +   coord_flip()          

Save plot

The ggsave() function will save the most recent plot in your current working directory unless you specify a path to another folder:

            ggplot(dat) +   aes(x = displ, y = hwy) +   geom_point()  ggsave("plot1.pdf")          

You can also specify the width, height and resolution (dpi) as follows:

            ggsave("plot1.pdf",   width = 12,   height = 12,   units = "cm",   dpi = 300 )          

Managing dates

If the time variable in your dataset is in date format, the {ggplot2} package recognizes the date format and automatically uses a specific type for the axis ticks.

There is no time variable with a date format in our dataset, so let's create a new variable of this type thanks to the as.Date() function:

            dat$date <- as.Date("2020-08-21") - 0:(nrow(dat) - 1)          

See the first 6 observations of this date variable and its class:

            head(dat$date)          
            ## [1] "2020-08-21" "2020-08-20" "2020-08-19" "2020-08-18" "2020-08-17" ## [6] "2020-08-16"          
            str(dat$date)          
            ##  Date[1:234], format: "2020-08-21" "2020-08-20" "2020-08-19" "2020-08-18" "2020-08-17" ...          

The new variable date is correctly specified in a date format.

Most of the time, with a time variable, we want to create a line plot with the date on the X-axis and another continuous variable on the Y-axis, like the following plot for example:

            p <- ggplot(dat) +   aes(x = date, y = hwy) +   geom_line() p          

As soon as the time variable is recognized as a date, we can use the scale_x_date() layer to change the format displayed on the X-axis. The following table shows the most frequent date formats:

Source: www.statmethods.net

Run ?strptime() to see many more date formats available in R.

For this example, let's add the year in addition to the unabbreviated month:

            p + scale_x_date(date_labels = "%B %Y")          

It also possible to control the breaks to display on the X-axis with the date_breaks argument. For this example, let's say we want to display the day as number and the abbreviated month for each interval of 10 days:

            p + scale_x_date(date_breaks = "10 days", date_labels = "%d %b")          

If labels displayed on the X-axis are unreadable because they overlap each other, you can rotate them with the theme() layer and the angle argument:

            p + scale_x_date(date_breaks = "10 days", date_labels = "%d %b") +   theme(axis.text.x = element_text(angle = 60, hjust = 1))          

Draw Regression Line on Plot

Source: https://statsandr.com/blog/graphics-in-r-with-ggplot2/