在 R 中将 cumsum 值重置为零

Question

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

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)}))

但它不起作用。

Answer 1

这是一个方法。基于 ave :

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

哪里 dat是您的数据框的名称。