R 中值的累积计数

我希望你做得很好。我想知道如何计算特定条件下数据集的累积和。我的数据集的简化版本如下所示：

t   id  
A   22
A   22
R   22
A   41
A   98
A   98
A   98
R   98
A   46
A   46
R   46
A   46
A   46
A   46
R   46
A   46
A   12
R   54
A   66
R   13 
A   13
A   13
A   13
A   13
R   13
A   13
  

想要创建一个新的数据集，其中对于“id”的每个值，我将获得每个 id 出现的累积次数，但是当 t=R 时，我需要重新启动计数，例如

t   id  count
A   22  1
A   22  2
R   22  0
A   41  1
A   98  1
A   98  2
A   98  3
R   98  0
A   46  1
A   46  2
R   46  0
A   46  1
A   46  2
A   46  3
R   46  0
A   46  1
A   12  1
R   54  0
A   66  1
R   13  0
A   13  1
A   13  2
A   13  3
A   13  4
R   13  0
A   13  1
  

关于如何做到这一点有什么想法吗？提前致谢。

Using rle:

out <- transform(df, count = sequence(rle(do.call(paste, df))$lengths))
out$count[out$t == "R"] <- 0

If your data.frame不止这两列，而您只想检查这两列，那么，只需替换df with df[, 1:2] (or) df[, c("t", "id")].

如果你发现do.call(paste, df)危险（如 @flodel 评论），那么你可以将其替换为：

as.character(interaction(df))

我个人认为这种设置没有任何危险或笨拙的地方（只要你有正确的分隔符，这意味着你很了解你的数据）。但是，如果您确实发现了这种情况，第二个解决方案可能会对您有所帮助。

Update:

对于那些不喜欢使用的人do.call(paste, df) or as.character(interaction(df))（请参阅我、@flodel 和 @hongOoi 之间的评论交流），这是另一个基本解决方案：

idx <- which(df$t == "R")
ww <- NULL
if (length(idx) > 0) {
    ww <- c(min(idx), diff(idx), nrow(df)-max(idx))
    df <- transform(df, count = ave(id, rep(seq_along(ww), ww), 
                   FUN=function(y) sequence(rle(y)$lengths)))
    df$count[idx] <- 0
} else {
    df$count <- seq_len(nrow(df))
}