4

我想对这些数据进行分组,但在分组时对某些列应用不同的函数。

ID  type isDesc isImage
1   1    1      0
1   1    0      1
1   1    0      1
4   2    0      1
4   2    1      0
6   1    1      0
6   1    0      1
6   1    0      0

我想按ID, 列分组isDesc并且isImage可以求和,但我想按原样获取类型的值。type在整个数据集中都是一样的。结果应如下所示:

ID  type isDesc isImage
1   1    1      2
4   2    1      1
6   1    1      1

目前我正在使用

library(plyr)
summarized = ddply(data, .(ID), numcolwise(sum))

但它只是总结了所有的列。你不必使用ddply,但如果你认为它对工作有好处,我想坚持下去。data.table图书馆也是一种选择

4

1 回答 1

6

使用data.table

require(data.table)
dt <- data.table(data, key="ID")
dt[, list(type=type[1], isDesc=sum(isDesc), 
                  isImage=sum(isImage)), by=ID]

#    ID type isDesc isImage
# 1:  1    1      1       2
# 2:  4    2      1       1
# 3:  6    1      1       1

使用plyr

ddply(data , .(ID), summarise, type=type[1], isDesc=sum(isDesc), isImage=sum(isImage))
#   ID type isDesc isImage
# 1  1    1      1       2
# 2  4    2      1       1
# 3  6    1      1       1

编辑:使用data.table's .SDcols,你可以这样做,以防你有太多的列要求和,而其他列只取第一个值。

dt1 <- dt[, lapply(.SD, sum), by=ID, .SDcols=c(3,4)]
dt2 <- dt[, lapply(.SD, head, 1), by=ID, .SDcols=c(2)]
> dt2[dt1]
#    ID type isDesc isImage
# 1:  1    1      1       2
# 2:  4    2      1       1
# 3:  6    1      1       1

您可以提供列名或列号作为 .SDcols 的参数。例如:.SDcols=c("type")也是有效的。

于 2013-03-15T13:54:13.227 回答