在R中,当不是所有因素组合都存在时,如何使用“聚合”或“通过”?

时间:2011-10-05 19:34:51

标签: r aggregate

这是一个用来说明我的数据的小例子:

> df <- data.frame(subgroup=rep(paste("s",1:3, sep=""), times=3),
                   feature=c(rep("a",6), rep("b",3)),
                   var=rep(1:3, each=3),
                   data=c(rnorm(3,1), rnorm(3,2), rnorm(3,0)))
> df
  subgroup feature var        data
1       s1       a   1  1.53152620
2       s2       a   1  1.25476445
3       s3       a   1  1.04221040
4       s1       a   2  1.68913400
5       s2       a   2  1.48290273
6       s3       a   2  1.62871854
7       s1       b   3  0.05278296
8       s2       b   3 -0.66623654
9       s3       b   3 -1.40006454

我想检查数据集中存在的feature-var的每个组合的“data”列的总和。更准确地说,我想在总和大于3时获得TRUE,否则为FALSE:

> result
  feature snp   res
1       a   1  TRUE
2       a   2  TRUE
3       b   3 FALSE

我尝试使用“聚合”或“按”,但不能使它们符合我的需要。任何的想法?提前谢谢。

1 个答案:

答案 0 :(得分:3)

一种方法是使用plyr的函数ddply对feature和var进行分组。您可以使用summarize功能创建一个新的data.frame,其中包含与您开发的规则相对应的列。

library(plyr)
ddply(df, c("feature", "var"), summarize, res = ifelse(sum(data) > 3,TRUE, FALSE))

结果:

  feature var   res
1       a   1  TRUE
2       a   2  TRUE
3       b   3 FALSE

另一种选择是使用data.table,这应该会带来一些性能上的好处:

library(data.table)
dt <- data.table(df)

dt[, ifelse(sum(data) > 3, TRUE, FALSE), by = c("feature", "var")]

     feature var    V1
[1,]       a   1  TRUE
[2,]       a   2  TRUE
[3,]       b   3 FALSE