在具有多个因素的 data.frame 中折叠变量时避免嵌套应用

Question

我有一个包含多个因素和多个数字变量的数据框。我想折叠其中一个因素（平均说）。

在我的尝试中，我只能想到嵌套的 sapply 或 for 循环来隔离要平均的数字元素。

var <- data.frame(A = c(rep('a',8),rep('b',8)), B = 
c(rep(c(rep('c',2),rep('d',2)),4)), C = c(rep(c('e','f'),8)),
                  D = rnorm(16), E = rnorm(16))
> var
   A B C           D           E
1  a c e  1.1601720731 -0.57092435
2  a c f -0.0120178626  1.05003748
3  a d e  0.5311032778  1.67867806
4  a d f -0.3399901000  0.01459940
5  a c e -0.2887561691 -0.03847519
6  a c f  0.0004299922 -0.36695879
7  a d e  0.8124655890  0.05444033
8  a d f -0.3777058654  1.34074427
9  b c e  0.7380720821  0.37708543
10 b c f -0.3163496271  0.10921373
11 b d e -0.5543252191  0.35020193
12 b d f -0.5753686426  0.54642790
13 b c e -1.9973216646  0.63597405
14 b c f -0.3728926714 -3.07669300
15 b d e -0.6461596329 -0.61659041
16 b d f -1.7902722068 -1.06761729


sapply(4:ncol(var), function(i){
  sapply(1:length(levels(var$A)), function(j){
    sapply(1:length(levels(var$B)), function(t){
      sapply(1:length(levels(var$C)), function(z){
        mean(var[var$A == levels(var$A)[j] & 
var$B == levels(var$B)[t] & 
var$C == levels(var$C)[z],i])
      })
    })
  })
})

             [,1]       [,2]
[1,]  0.435707952 -0.3046998
[2,] -0.005793935  0.3415393
[3,]  0.671784433  0.8665592
[4,] -0.358847983  0.6776718
[5,] -0.629624791  0.5065297
[6,] -0.344621149 -1.4837396
[7,] -0.600242426 -0.1331942
[8,] -1.182820425 -0.2605947

有没有办法在没有这么多申请的情况下做到这一点？也许用 mapply 或 outer

Answer 1

也许只是，

var <- data.frame(A = c(rep('a',8),rep('b',8)), B = 
                    c(rep(c(rep('c',2),rep('d',2)),4)), C = c(rep(c('e','f'),8)),
                  D = rnorm(16), E = rnorm(16))

library(dplyr)
var %>%
  group_by(A,B,C) %>%
  summarise_if(is.numeric,mean)

（请注意，当我运行你的申请代码时，你显示的输出不是我得到的，但上面的内容与我运行你的申请代码时得到的相同。）

Answer 2

对于内联聚合（保持相同的数据帧行数），考虑ave:

var$D_mean <- with(var, ave(D, A, B, C, FUN=mean))
var$E_mean <- with(var, ave(E, A, B, C, FUN=mean))

对于完整聚合（折叠为因子组），请考虑 aggregate:

aggregate(. ~ A + B + C, var, mean)

Answer 3

我将用data.table解决方案来完成三位一体。这里 .SD 是 by 部分未列出的所有列的 data.table。这是对 this question 的近乎重复（唯一的区别是 >1 列被汇总），所以如果您需要更多解决方案，请单击它。

library(data.table)
setDT(var)

var[, lapply(.SD, mean), by = .(A, B, C)]
#    A B C           D            E
# 1: a c e  0.07465822  0.032976115
# 2: a c f  0.40789460 -0.944631574
# 3: a d e  0.72054938  0.039781185
# 4: a d f -0.12463910  0.003363382
# 5: b c e -1.64343115  0.806838905
# 6: b c f -1.08122890 -0.707975411
# 7: b d e  0.03937829  0.048136471
# 8: b d f -0.43447899  0.028266455

在具有多个因素的 data.frame 中折叠变量时避免嵌套应用

avoiding nested sapply when collapsing variable in data.frame with multiple factors

for-loop

r

sapply