More Purrr for GGPlot2

[jdnewmil]

Thanks to http://r4ds.had.co.nz/ I can riff on your idea and point out that the "do" function doesn't play well with other columns. A little more purrr and some tidyr and you can work with many list columns.

library(ggpmisc) # for annotation
library(tidyr) # for nest
plots <- small_pwt %>%
         mutate( country = as.character( country ) ) %>%
         group_by(country) %>%
         nest %>%
         mutate( avhlm = map( data
                            , function( DF ) {
                                lm( avh ~ year, data = DF )
                              }
                            )
               ) %>%
         mutate( plot = pmap( .
                            , function( country, data, avhlm ) { 
                                ggplot( data, aes( y = avh, x = year ) ) + 
                                  theme_tufte() +
                                  geom_line() +
                                  geom_smooth( method = "lm" ) +
                                  stat_poly_eq( formula = y ~ x
                                              , aes( label = paste( ..eq.label..
                                                                  , ..rr.label..
                                                                  , sep = "*\",\"~~~"
                                                                  )
                                                   )
                                              , label.y.npc = "bottom"
                                              , parse = TRUE
                                              ) +
                                  ggtitle( country ) +
                                  ylab( "Year") +
                                  xlab( "Average annual hours worked by persons engaged")
                              }
                            )
               )

# RStudio keeps a history you can surf through
walk( plots$plot, print )

[b-rodrigues]

Hey thanks for the message! For some reason I've completely forgot to answer, and then I forgot all about it :/ Shame on me!

In any case thanks, I'll study your code, there's some stuff I don't understand yet (like stat_poly_eq, and everything that follows... seems useful!)