修改 lm 或 loess 函数以在 ggplot2 的 geom_smooth 中使用它

Question

我需要修改 lm （或最终 loess）函数，以便我可以在 ggplot2 的 geom_smooth （或 stat_smooth）中使用它>）。

例如，这就是 stat_smooth 的正常使用方式：

> qplot(data=diamonds, carat, price, facets=~clarity) + stat_smooth(method='lm')`

我想定义一个自定义 lm2 函数，用作 method 参数的值stat_smooth，这样我就可以自定义它的行为。

> lm2 <- function(formula, data, ...)
  {
      print(head(data))
      return(lm(formula, data, ...))
  }
> qplot(data=diamonds, carat, price, facets=~clarity) + stat_smooth(method='lm2')

请注意，我在 stat_smooth 中使用了 method='lm2' 作为参数。 当我执行此代码时出现错误：

eval(expr, envir, enclos) 中的错误：“nthcdr”需要一个 CDR 列表

，我不太明白。在 stat_smooth 之外运行时，lm2 方法运行良好。我玩了一下，遇到了不同类型的错误，但由于我对 R 的调试工具不太熟悉，所以很难调试它们。老实说，我不明白应该在 return() 调用中放入什么内容。

Answer 1

在函数调用中使用 ... 作为参数有一些奇怪的地方，我不完全理解（它与 ... 作为列表类型对象有关）。

这是一个版本，它的工作原理是将函数调用作为对象，将要调用的函数设置为 lm，然后在我们自己的调用者的上下文中评估该调用。计算的结果就是我们的返回值（在 R 中，函数中最后一个表达式的值就是返回值，因此我们不需要显式的返回值）。

foo <- function(formula,data,...){
   print(head(data))
   x<-match.call()
   x[[1]]<-quote(lm)
   eval.parent(x)
}

如果你想在 lm 调用中添加参数，你可以这样做：

x$na.action <- 'na.exclude'

如果你想在调用 lm 之前删除 foo 的参数，你可以这样做

x$useless <- NULL

顺便说一句，geom_smooth 和stat_smooth 将任何额外参数传递给平滑函数，因此如果您只需要设置一些额外参数，则无需创建自己的函数

qplot(data=diamonds, carat, price, facets=~clarity) + 
  stat_smooth(method="loess",span=0.5)