寻找一种更优雅的方式来填充具有列表中特定值的矩阵

时间:2015-07-03 19:44:45

标签: r for-loop

我有一个相当复杂的结构列表,我想从中提取一些特定的值。由于数据的dput输出和列表结构在这种情况下似乎有问题,因此我在此处重现用户生成此列表的@ user34771代码posted yesterday

set.seed(123)
testdf <- data.frame(vy = rnorm(60), vx = rnorm(60) , gvar = rep(c("a","b"), each=30))
require(fBasics)
normfuns <- list(jarqueberaTest=jarqueberaTest, shapiroTest=shapiroTest, lillieTest=lillieTest)
mynormtest <- function(d) {
norm_test <- res_reg <- list()
for (i in c("a","b")){
  res_reg[[i]] <- residuals(lm(vy~vx, data=d[d$gvar==i,]))
  norm_test[[i]] <- lapply(normfuns, function(f) f(res_reg[[i]]))  
 }
return(norm_test) 
}
res_list <- mynormtest(testdf)

对于res_list中的数据,我有一个函数可以提取我感兴趣的值并将它们存储在矩阵中:

getparams2 <- function(myp) {
m <- matrix(NA, nrow=length(myp), ncol=3)
for (i in (1:length(myp))){
 m[i,] <- sapply(1:3,function(x) myp[[i]][[x]]@test$statistic)}
return(m)
}

此功能提供所需的输出。

> getparams2(res_list)
          [,1]      [,2]       [,3]
[1,] 0.9917054 0.9670860 0.08837058
[2,] 0.1421776 0.9806379 0.12619274

但是,我对函数getparams2()及其FOR循环的样式不满意。我很高兴知道这个功能是否可以用更优雅,更紧凑的形式编写。谢谢你的帮助。

1 个答案:

答案 0 :(得分:2)

获得矩阵:

t(sapply(res_list, sapply, function(x) x@test$statistic))
#  jarqueberaTest.X-squared shapiroTest.W lillieTest.D
#a                0.9917054     0.9670860   0.08837058
#b                0.1421776     0.9806379   0.12619274

rapply

t(simplify2array(rapply(res_list, function(x) x@test$statistic, how = "list")))
#  jarqueberaTest shapiroTest lillieTest
#a 0.9917054      0.967086    0.08837058
#b 0.1421776      0.9806379   0.1261927