2

我需要编写一个包含三个参数的函数：dat - 数据框的名称，mainVar - 用于计算第三个变量之间相关性的字符向量，以及 varlist - 包含一个或多个值的字符向量。

该函数将返回一个数据框，其中包含每对之间的相关系数和相应的 p 值。

我希望实现的一个例子：

 myCortest (chol, "wt", "age")
     var1  var2          R            p
 age   wt   age  0.6660014 5.631448e-26

到目前为止我所拥有的：

myCortest <- function(dat, mainVar, varlist){
result <- data.frame()
for (i in 1:length(mainVar)){
foo <- cor.test(dat$mainvar, dat$varlist)
r <- data.frame(Varname = mainVar[i],
R <- as.vector(foo$estimate[1]),
P <- foo$p.value)
result <- rbind(result, r)
}
return(result)
}

我的代码不会运行，所以我知道我做错了什么。我怎样才能达到我想要的输出？

4

1 回答 1

1

如果我的目标正确，这是一个 sa(i)mple 函数：

myCortest <- function(dat, mainVar, varlist){

foo<-lapply(varlist,function(x){foo1<-cor.test(get(mainVar,as.environment(dat)), 
                       get(x,as.environment(dat)))
            data.frame(Var1=mainVar,Var2=x,
                       p.value=foo1$p.value,R.Sq=foo1$estimate)
            })

foo
}

测试它：

myCortest (iris, "Sepal.Length", c("Petal.Length","Sepal.Width"))

输出：

[[1]]
            Var1         Var2      p.value      R.Sq
cor Sepal.Length Petal.Length 1.038667e-47 0.8717538

[[2]]
            Var1        Var2   p.value       R.Sq
cor Sepal.Length Sepal.Width 0.1518983 -0.1175698
于 2019-03-10T07:12:39.827 回答