我想要一个函数,其中参数可以采用变量名称(它是数据集的一部分,但不作为对象存储在环境中)并将该变量名称插入模型公式中。
例如:
# Some data with a couple of variables
my_df <- data.frame(y = rbinom(10, 1,0.5), var1 = runif(10), var2 = runif(10))
# A function that fits a model using predictor specified in the arguments
my_fun <- function(var_name, df){
glm(y ~ var_name, data = df, family = "binomial")
}
当我尝试使用该功能时,我收到以下错误消息
my_fun(var1, my_df)
Error in eval(expr, envir, enclos) : object 'var1' not found
# What I want the function to do
glm(y ~ var1, data = my_df, family = "binomial")
有没有办法让这种功能发挥作用?
您可以使用以下命令解析未var_name
加引号substitute
:
my_fun <- function(var_name, df){
glm.formula <- substitute(y ~ x, list(x = substitute(var_name)))
glm(glm.formula, data = df, family = "binomial")
}
一个例子:
my_fun(var1, my_df)
# Call: glm(formula = glm.formula, family = "binomial", data = df)
#
# Coefficients:
# (Intercept) var1
# -1.226 3.108
#
# Degrees of Freedom: 9 Total (i.e. Null); 8 Residual
# Null Deviance: 13.46
# Residual Deviance: 11.35 AIC: 15.35
glm(y ~ var1, data = my_df, family = "binomial")
# Call: glm(formula = y ~ var1, family = "binomial", data = my_df)
#
# Coefficients:
# (Intercept) var1
# -1.226 3.108
#
# Degrees of Freedom: 9 Total (i.e. Null); 8 Residual
# Null Deviance: 13.46
# Residual Deviance: 11.35 AIC: 15.35
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句