R中的累计数

时间:2012-04-05 13:09:25

标签: r count cumulative-sum

有没有办法计算对象在R中累积出现在列中的次数?

e.g。说我有专栏:

id  
1  
2  
3  
2  
2  
1  
2  
3

这将成为:

id   count  
1     1  
2     1  
3     1  
2     2  
2     3  
1     2  
2     4  
3     2  

等...

由于

6 个答案:

答案 0 :(得分:28)

ave函数按组计算函数。

> id <- c(1,2,3,2,2,1,2,3)
> data.frame(id,count=ave(id==id, id, FUN=cumsum))
  id count
1  1     1
2  2     1
3  3     1
4  2     2
5  2     3
6  1     2
7  2     4
8  3     2

我使用id==id创建所有TRUE值的向量,这些值在传递给cumsum时会转换为数字。您可以将id==id替换为rep(1,length(id))

答案 1 :(得分:9)

以下是获取计数的方法:

id <- c(1,2,3,2,2,1,2,3)

sapply(1:length(id),function(i)sum(id[i]==id[1:i]))

这给了你:

[1] 1 1 1 2 3 2 4 2

答案 2 :(得分:3)

我的数据框太大,接受的答案一直在崩溃。这对我有用:

library(plyr)
df$ones <- 1
df <- ddply(df, .(id), transform, cumulative_count = cumsum(ones))
df$ones <- NULL 

答案 3 :(得分:3)

dplyr方式:

library(dplyr)

foo <- data.frame(id=c(1, 2, 3, 2, 2, 1, 2, 3))
foo <- foo %>% group_by(id) %>% mutate(count=row_number())
foo

# A tibble: 8 x 2
# Groups:   id [3]
     id count
  <dbl> <int>
1     1     1
2     2     1
3     3     1
4     2     2
5     2     3
6     1     2
7     2     4
8     3     2

最后以id分组。如果不希望将其分组,请添加%>% ungroup()

答案 4 :(得分:2)

出于完整性考虑,添加data.table方式:

library(data.table)

DT <- data.table(id = c(1, 2, 3, 2, 2, 1, 2, 3))

DT[, count := seq(.N), by = id][]

输出:

   id count
1:  1     1
2:  2     1
3:  3     1
4:  2     2
5:  2     3
6:  1     2
7:  2     4
8:  3     2

答案 5 :(得分:0)

用于获取任何数组(包括非数字数组)的累积计数的函数:

cumcount <- function(x){
  cumcount <- numeric(length(x))
  names(cumcount) <- x

  for(i in 1:length(x)){
    cumcount[i] <- sum(x[1:i]==x[i])
  }

  return(cumcount)
}