25
votes

I would like to create one separate plot per group in a data frame and include the group in the title.

With the iris dataset I can in base R and ggplot do this

plots1 <- lapply(split(iris, iris$Species), 
  function(x) 
    ggplot(x, aes(x=Petal.Width, y=Petal.Length)) +
      geom_point() +
      ggtitle(x$Species[1]))

Is there an equivalent using dplyr?

Here's an attempt using facets instead of title.

p <- ggplot(data=iris, aes(x=Petal.Width, y=Petal.Length)) + geom_point()
plots2 = iris %>% group_by(Species) %>% do(plots = p %+% . + facet_wrap(~Species))

where I use %+% to replace the dataset in p with the subset for each call.

Workaround with facets

or (working but complex) with ggtitle

plots3 = iris %>%
  group_by(Species) %>%
  do(
    plots = ggplot(data=.) +
      geom_point(aes(x=Petal.Width, y=Petal.Length)) +
      ggtitle(. %>% select(Species) %>% mutate(Species=as.character(Species)) %>% head(1) %>% as.character()))

Working example

The problem is that I can't seem to set the title per group with ggtitle in a very simple way.

Thanks!

3

3 Answers

44
votes

Use .$Species to pull the species data into ggtitle:

iris %>% group_by(Species) %>% do(plots=ggplot(data=.) +
         aes(x=Petal.Width, y=Petal.Length) + geom_point() + ggtitle(unique(.$Species)))
7
votes

From dplyr 0.8.0 we can use group_map :

library(dplyr, warn.conflicts = FALSE, quietly = TRUE)
#> Warning: le package 'dplyr' a été compilé avec la version R 3.5.2
library(ggplot2)
plots3 <- iris %>%
  group_by(Species) %>%
  group_map(~tibble(plots=list(
    ggplot(.) + aes(x=Petal.Width, y=Petal.Length) + geom_point() + ggtitle(.y[[1]]))))

plots3
#> # A tibble: 3 x 2
#> # Groups:   Species [3]
#>   Species    plots   
#>   <fct>      <list>  
#> 1 setosa     <S3: gg>
#> 2 versicolor <S3: gg>
#> 3 virginica  <S3: gg>
plots3$plots[[2]]

Created on 2019-02-18 by the reprex package (v0.2.0).

2
votes

This is another option using rowwise:

plots2 = iris %>% 
    group_by(Species) %>% 
    do(plots = p %+% .) %>% 
    rowwise() %>%
    do(x=.$plots + ggtitle(.$Species))