将 R 中的 cumsum 值重置为零

我有个问题。例如我有这个：

id  truth count
1   1     1
2   1     2
3   0     0
4   1     1
5   1     2
6   1     3
8   0     0

我试过这个：

fun <- rle(df$truth)

df$count <- unlist(sapply(fun$lengths, function(x) {return(1:x)}))

但它不起作用。

这是一个方法。基于ave:

transform(dat, count = truth * ave(truth, c(0L, cumsum(diff(truth) != 0)), 
                                   FUN = seq_along))

where dat是您的数据框的名称。