在R中的数据表的by =调用中循环编号的变量

Question

这可能很愚蠢，但是我找不到以下问题的（非详尽的）解决方法：

让我们设置：

data<-data.table(id=c("a","a","a","b","b"),
                 x1=1:5,
                 x2=6:10,
                 x3=11:15)

我想说，根据“ id”分组的每个列变量的均值，然后将它们作为变量附加到数据集。 当然，只有很少的几个，我们可以使用以下语法：

means1<-data[,mean(x1),by=id]

同样，对于x2，x3，可以得到：

   id x1 x2 x3  V1 V1.1 V1.2
1:  a  1  6 11 2.0  7.0 12.0
2:  a  2  7 12 2.0  7.0 12.0
3:  a  3  8 13 2.0  7.0 12.0
4:  b  4  9 14 4.5  9.5 14.5
5:  b  5 10 15 4.5  9.5 14.5

但是，对于更大的集合，我们很想循环。 这是我首先尝试的方法：

for(nn in 1:3){
   data<-data[data[,mean(paste("x",nn,sep="")),by=id]]
}

但这失败了，我想是因为即使通常允许使用字符串来标识列名，但mean函数也会尝试在外部运算符之前进行操作：

Warning messages:
1: In `[.data.table`(data, , mean(paste("x", nn, sep = "")), by = id) :
  argument is not numeric or logical: returning NA

因此，下一个尝试：

for(nn in 1:3){
   data<-data[data[,mean(data[[paste("x",nn,sep="")]]),by=id]]
}

但是，这也失败了，这也是因为均值首先起作用。 （更不用说语法对于在STATA中曾经做过类似事情的任何人来说都是值得的）因此，我们最终得到了每个x分配给每个id值的总体平均值：

   id x1 x2 x3 V1 V1.1 V1.2
1:  a  1  6 11  3    8   13
2:  a  2  7 12  3    8   13
3:  a  3  8 13  3    8   13
4:  b  4  9 14  3    8   13
5:  b  5 10 15  3    8   13

那么，我们将如何循环执行此简单任务？

这样的语法在类似的调用中对我有用，例如：

for(nn in 1:3){
   data[,paste("x_greater_than_4_",nn,sep=""):=(data[[paste("x",nn,sep="")]]>4)]
}

产生我所期望的：

   id x1 x2 x3 x_greater_than_4_1 x_greater_than_4_2 x_greater_than_4_3
1:  a  1  6 11              FALSE               TRUE               TRUE
2:  a  2  7 12              FALSE               TRUE               TRUE
3:  a  3  8 13              FALSE               TRUE               TRUE
4:  b  4  9 14              FALSE               TRUE               TRUE
5:  b  5 10 15               TRUE               TRUE               TRUE

Answer 1

我首先要使用setkey设置一个键，然后在j表达式中使用lapply并自连接结果。 您可以在lapply使用.SD以及关联的.SDcols以数字位置指定列。 像这样：

setkey( data , id )
data[ data[ , lapply( .SD , mean ) , keyby = id , .SDcols = 2:4 ] ]
#   id x1 x2 x3 x1.1 x2.1 x3.1
#1:  a  1  6 11  2.0  7.0 12.0
#2:  a  2  7 12  2.0  7.0 12.0
#3:  a  3  8 13  2.0  7.0 12.0
#4:  b  4  9 14  4.5  9.5 14.5
#5:  b  5 10 15  4.5  9.5 14.5

# If you just want the group means use this:
data[ ,  lapply( .SD , mean ), by = id , .SDcols = 2:4 ]

---

另外，您可以按如下方式使用:=以及by ，这也将避免连接：

sd_cols = c("x1", "x2", "x3")
data[, c(paste0("v", 1:3)) := lapply(.SD, mean), by=id, .SDcols=sd_cols]

#    id x1 x2 x3  v1  v2   v3
# 1:  a  1  6 11 2.0 7.0 12.0
# 2:  a  2  7 12 2.0 7.0 12.0
# 3:  a  3  8 13 2.0 7.0 12.0
# 4:  b  4  9 14 4.5 9.5 14.5
# 5:  b  5 10 15 4.5 9.5 14.5

Answer 2

尝试在公式接口中使用aggregate来获取组均值，然后merge到原始数据中：

merge(data,aggregate(.~id,data=data,mean),by="id",suffixes=c("",".mean"))
   id x1 x2 x3 x1.mean x2.mean x3.mean
1:  a  1  6 11     2.0     7.0    12.0
2:  a  2  7 12     2.0     7.0    12.0
3:  a  3  8 13     2.0     7.0    12.0
4:  b  4  9 14     4.5     9.5    14.5
5:  b  5 10 15     4.5     9.5    14.5

Answer 3

怎么样：

> data[, x1Mean := mean(x1), by=id] # this command updates the data table
> data
   id x1 x2 x3 x1Mean
1:  a  1  6 11    2.0
2:  a  2  7 12    2.0
3:  a  3  8 13    2.0
4:  b  4  9 14    4.5
5:  b  5 10 15    4.5