按组平均,删除当前行

sda*_*aza 4 r data.table

我想计算变量的组均值,但不包括焦点受访者:

set.seed(1)
dat <- data.table(id = 1:30, y = runif(30), grp = rep(1:3, each=10))
Run Code Online (Sandbox Code Playgroud)

第一个记录(受访者)应平均为...第二个记录,依此类推:

mean(dat[c==1, y][-1])
mean(dat[c==1, y][-2])
mean(dat[c==1, y][-3])
Run Code Online (Sandbox Code Playgroud)

对于第二组相同:

mean(dat[c==2, y][-1])
mean(dat[c==2, y][-2])
mean(dat[c==2, y][-3])
Run Code Online (Sandbox Code Playgroud)

我试过了,但是没有用:

ex[, avg := mean(ex[, y][-.I]), by=grp]
Run Code Online (Sandbox Code Playgroud)

有任何想法吗?

Mar*_*pov 5

您可以尝试以下解决方案:

set.seed(1)
dat <- data.table(id = 1:9, y = c(NA,runif(8)), grp = rep(1:3, each=3))

dat[, avg2 := sapply(seq_along(y),function(i) mean(y[-i],na.rm=T)), by=grp]

dat
#    id         y grp      avg2
# 1:  1        NA   1 0.3188163
# 2:  2 0.2655087   1 0.3721239
# 3:  3 0.3721239   1 0.2655087
# 4:  4 0.5728534   2 0.5549449
# 5:  5 0.9082078   2 0.3872676
# 6:  6 0.2016819   2 0.7405306
# 7:  7 0.8983897   3 0.8027365
# 8:  8 0.9446753   3 0.7795937
# 9:  9 0.6607978   3 0.9215325
Run Code Online (Sandbox Code Playgroud)

  • 随着数据量的增加,这种双重循环(如果包含“ by”循环,则为三倍循环)将成倍地缓慢。 (3认同)