我想定义一个函数,当我输入一个字符串作为协变量时,该函数会将我的字符串放在特定位置并将其转换为公式。我知道我的代码不正确,但我不知道如何编写。
我想要的是,当我输入covars <- "+s(time,bs= 'cr',fx=TRUE,k=7)"该函数时,该函数将添加covars到这样的公式中gam.model <- gam(cvd ~ pm10 +s(time,bs= 'cr',fx=TRUE,k=7), data = chicagoNMMAPS , family =poisson, na.rm=T)
library(dlnm) # use chicagoNMMAPS data
library(mgcv)
# define myfun
myfun <- function(covars){
covars <- covars
gam.model <- gam(cvd ~ pm10 + covars, data = chicagoNMMAPS , family =poisson, na.rm=T)
summary(gam.model)
}
myfun("+s(time,bs= 'cr',fx=TRUE,k=7)")
Run Code Online (Sandbox Code Playgroud)
myfun 应该这样做:
gam.model <- gam(cvd ~ pm10 + covars, data = chicagoNMMAPS , family =poisson, na.rm=T)
Run Code Online (Sandbox Code Playgroud)
您是否在寻找这个,不确定,但尝试as.formula一下paste0:
myfunc_formula <- function(covars){
return(as.formula(paste0('cvd ~ pm10 ', covars)))
}
Run Code Online (Sandbox Code Playgroud)
我们稍后可以使用这个输入gam(myfunc_formula(covars), data = chicagoNMMAPS , family =poisson, na.rm=T),
## In case someone wants to return the summary of given gam model
myfunc_formula_v1 <- function(covars){
gam1 <- gam(as.formula(paste0('cvd ~ pm10 ', covars)), data = chicagoNMMAPS , family =poisson, na.rm=TRUE)
return(summary(gam1))
}
Run Code Online (Sandbox Code Playgroud)
我们还可以通过提供目标变量名称等输入参数来使其灵活。
例如另一个版本可能是:
myfunc_formula_v2 <- function(covars, target='cvd'){
return(as.formula(paste0(target, ' ~ pm10 ', covars)))
}
Run Code Online (Sandbox Code Playgroud)
输出:
> myfunc_formula(covars)
cvd ~ pm10 + s(time, bs = "cr", fx = TRUE, k = 7)
Run Code Online (Sandbox Code Playgroud)
给定covars = "+s(time,bs= 'cr',fx=TRUE,k=7)"