R: for 循环与 data.table 库

Question

我使用的library(data.table)我愿做一for循环计算的mean列1:4 ，根据col5值。

colnames(DT) # "col1" "col2" "col3" "col4" "col5" 
for (i in 1:4){
  o=colnames(DT)[[i]]
  l=DT[,mean(o), by=col5]
  print(l)
}

问题是DT不将colnames作为character向量（“col”）。 任何建议表示赞赏。

Answer 1

阅读 data.table 小插图。

library(data.table)
set.seed(42)
DT <- data.table(matrix(rnorm(100), ncol=4))
setnames(DT, paste0("col", 1:4))
DT[, col5 := rep(1:5, 5)]

DT[, lapply(.SD, mean), by = col5]

DT[, lapply(.SD, mean), by = col5, .SDcols = paste0("col", 2:3)]

Answer 2

尝试：

DT[,list(mean1=mean(col1), mean2=mean(col2), mean3=mean(col3), mean4=mean(col4)),by=col5]

如果列太多：

attach(DT)    
for(i in 1:4) print(DT[,mean(get(colnames(DT)[i])),by=DT[,5]])