ggplot2 + aes_string 通过公式接口在函数内

Question

以交互方式，此示例运行良好：

p <- ggplot(mtcars, aes(mpg, wt)) + geom_point()
p + facet_grid(. ~ vs)

现在，用公式接口做一个函数，用aes_string做同样的事情，还是不行（错误是：Error in layout_base(data, cols, drop = drop) : At least one layer must contain all variables used for facetting）：

tf <- function(formula, data) {
res <- as.character(formula[[2]])
fac2 <- as.character(formula[[3]][3])
fac1 <- as.character(formula[[3]][2])

# p <- ggplot(aes_string(x = fac1, y = res), data = data)
# p <- p + geom_point() # original attempt
p <- ggplot() # This is Joran's trick, but it doesn't work here
p <- p + geom_point(aes_string(x = fac1, y = res), data = data)
p <- p + facet_grid(.~fac2) # comment this out, and it works but
# of course is not faceted
}

p <- tf(formula = wt ~ am*vs, data = mtcars)

通过 Joran 的技巧，我参考了 here，这是我最近发布的一个类似问题。在这种情况下，ggplot2没有看到我的分面请求。使其facet_grid(".~fac2") 无效。建议？我一直被这些事情弄得不知所措。谢谢！

Answer 1

您可以使用as.formula 和paste：

p <- p + facet_grid(as.formula(paste(". ~", fac2)))

在您的示例中，这给出了：

Answer 2

我使用formula.tools 来处理公式。

我认为你应该对分面公式有一个单独的参数，否则你将不得不创建自己的解析器来计算wt ~ am*vs 在分面方面的含义

类似点阵分组的想法可能有用

wt~am | vs

但您必须深入研究 lattice 才能了解它们如何解析公式（看看latticeParseFormula - 很复杂！）

将两个公式分开更容易。您可以将facet 参数的字符变量列表作为rhs 和lhs 传递给facet_grid

我还使用了environment = parent.frame()，它适用于我的小型测试

library(formula.tools)

tf <- function(formula, faceting = NULL, data, print = TRUE) {
   y <- rhs(formula)
   x <- lhs(formula)

  p <- ggplot(environment = parent.frame()) 

  p <- p + geom_point(aes_string(x = x, y = y), data = data) 
   if (! is.null(faceting)){
     rhsfacet <- all.vars(rhs(faceting))
     lhsfacet <- all.vars(lhs(faceting))
     if(length(lhsfacet)==1 & any(lhsfacet %in% '.')) {lhsfacet <- NULL}
     if(length(rhsfacet)==1 & any(rhsfacet %in% '.')) {rhsfacet <- NULL}
     p <- p+ facet_grid(facet = list( lhsfacet, rhsfacet))}
  if(print) {print(p)}
  p 

}
tf(wt~mpg, faceting = ~am, data = mtcars, print = TRUE)