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

时间:2010-03-03 11:12:29

标签: r ggplot2 lm

我需要修改lm(或最终loess)函数,以便我可以在ggplot2的geom_smooth(或stat_smooth)中使用它。

例如,这就是stat_smooth正常使用的方式:

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

我想定义一个自定义lm2函数,以用作methodstat_smooth参数的值,以便我可以自定义其行为。

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

请注意,我在method='lm2'中使用了stat_smooth作为参数。 当我执行此代码时,得到错误:

  

eval中的错误(expr,envir,enclos):'nthcdr'需要一个列表来转发

我不太了解。在lm2之外运行时,stat_smooth方法非常有效。我玩了一下,我有不同类型的错误,但由于我不熟悉R的调试工具,我很难调试它们。老实说,我没有把我应该放在return()电话里面。

1 个答案:

答案 0 :(得分:6)

使用...作为函数调用中的参数有些奇怪,我并不完全理解(它与...作为列表类型对象有关)。

这是一个通过将函数调用作为对象来工作的版本,将要调用的函数设置为lm,然后在我们自己的调用者的上下文中评估调用。此评估的结果是我们的返回值(在R中,函数中最后一个表达式的值是返回的值,因此我们不需要显式的return)。

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_smoothstat_smooth将任何额外的参数传递给平滑函数,因此如果只需要设置一些额外的参数,就不需要创建自己的函数

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