r - 按变量编号行，但在满足条件时重新开始

Question

我想对数据框中的某些行组合进行编号（按 ID 和时间排序）

tc <- textConnection('
id              time       end_yn
abc             10         0
abc             11         0
abc             12         1
abc             13         0
def             10         0
def             15         1
def             16         0
def             17         0
def             18         1
')

test <- read.table(tc, header=TRUE)

目标是创建一个新列（“ number”），每行编号id从1 to n直到end_yn == 1被击中。之后end_yn == 1，编号应该重新开始。

在不考虑end_yn == 1条件的情况下，可以使用以下方法对行进行编号：

DT <- data.table(test)
DT[, id := seq_len(.N), by = id]

然而，预期的结果应该是：

id              time       end_yn   number
abc             10         0        1
abc             11         0        2
abc             12         1        3 
abc             13         0        1 
def             10         0        1
def             15         1        2
def             16         0        1
def             17         0        2
def             18         1        3

如何结合end_yn == 1条件？

Answer 1

我猜有不同的方法可以做到这一点，但这里有一个：

DT[, cEnd := c(0,cumsum(end_yn)[-.N])] # carry the end value forward

DT[, number := seq_len(.N), by = "id,cEnd"] # create your sequence

DT[, cEnd := NULL] # remove the column created above

设置id为 keyDT可能值得。