jpinelo - 1 year ago 100
R Question

# Using functions from dplyr package to add equation to qqplot with facets

I'm trying to produce scatterplots with regression equation and r2 for grouped data.

I can do one, but with grouped data I'm finding trouble when calculating the equations and r2 for all groups in a way that can be automatically extracted and added as annotation.

I believe that I'm pretty close, just making some silly mistake but can't seem to identify it.

1 - First I create a function that creates a model and the string of characters with the results.

``````library(dplyr)
eqlabels <- function(iris){
m <- lm(Sepal.Length ~ Sepal.Width, iris);
eq <- substitute(italic(y) == a + b * italic(x) * "," ~~ italic(r) ^ 2 ~ "=" ~ r2,
list(a = format(coef(m)[1], digits = 3),
b = format(coef(m)[2], digits = 3),
r2 = format(summary(m)\$r.squared, digits = 2)))
as.character(as.expression(eq));
}
``````

I came as far as this, but on step 2 it all breaks down:

2 - Now I must use the function on the grouped data.

This post suggests the use of ddply (from plyr package). I tried to replace that with something equivalent from the dplyr package, as suggested here.

`````` labelsP3 <- iris %>% group_by(Species) %>% do(eqlabels(.))
``````

However, this results in warning message (and then it does not plot...):
Warning message:

Error: Results are not data frames at positions: 1, 2, 3

As suggested here, I tried:

``````labelsP3 <- iris %>% group_by(Species) %>% do(with(eqlabels(iris)))
``````

But this results in error:

Error in eval(substitute(expr), data, enclos = parent.frame()) :
invalid 'envir' argument of type 'character'

The plotting should be fine like this, but I'm stuck at this stage.

``````plot3 <- ggplot(iris, aes(x = Sepal.Length, y = Sepal.Width)) + geom_point(colour = "grey60") +
facet_grid(Species ~ .) +
stat_smooth(method = lm) +
annotate("text", label = labelsP3, parse = TRUE)
``````

Thank you.

Alrighty let's try this again:

do the following:`labelsP3<-ddply(iris,.(Species),eqlabels)` that will get you your equations:

``````    Species
1     setosa  italic(y) == "2.64" + "0.69" * italic(x) * ","
~italic(r)^2 ~ "=" ~ "0.55"
2 versicolor italic(y) == "3.54" + "0.865" * italic(x) * "," ~
~italic(r)^2 ~ "=" ~ "0.28"
3  virginica italic(y) == "3.91" + "0.902" * italic(x) * "," ~
~italic(r)^2 ~ "=" ~ "0.21"
``````

Now that you have the equations, you should easily be able to plot them on your graph

you can then use this to graph the equations on your plot

``````geom_text(data=labels3, aes(label=V1, x=7, y=2), parse=TRUE)
``````

EDIT: THIRD TIME IS A CHARM

So after a lots of trial and error I got it to work, I still get a warning but at least it's a step in the right direction. As I suspected earlier, you have to use `as.data.frame`, like so: `labelsP3 <- iris %>% group_by(Species) %>% do(as.data.frame(eqlabels(.)))`

you get the following output:

``````     Source: local data frame [3 x 2]
Groups: Species [3]

Species                                                                   eqlabels(.)
(fctr)                                                                         (chr)
1     setosa  italic(y) == "2.64" + "0.69" * italic(x) * "," ~
~italic(r)^2 ~ "=" ~ "0.55"
2 versicolor italic(y) == "3.54" + "0.865" * italic(x) * "," ~
~italic(r)^2 ~ "=" ~ "0.28"
3  virginica italic(y)

== "3.91" + "0.902" * italic(x) * "," ~ ~italic(r)^2 ~ "=" ~ "0.21"
``````

UPDATE:

For the plotting part you can do it as follow:

``````    plot3 <- ggplot(iris, aes(x = Sepal.Length, y = Sepal.Width)) + geom_point(colour = "grey60") +
facet_grid(Species ~ .) +
stat_smooth(method = lm) +
geom_text(data=labelsP3, aes(label=`eqlabels(.)`, x=7, y=2), parse=TRUE)
``````

the x and y is geom_text is for the placement of the label on the graph.

or this even looks a bit better:

`````` plot3 + geom_text(data=labelsP3, aes(label=`eqlabels(.)`, vjust = -1, +
hjust=-0.5,x=4, y=0), parse=TRUE)
``````

Recommended from our users: Dynamic Network Monitoring from WhatsUp Gold from IPSwitch. Free Download