R data.table當條件爲TRUE時，找到下N行

Question

如何在條件爲TRUE時獲得該行，並在它後面N行？如果N行中的任何一行的條件爲TRUE，請重新啓動計數。也爲每個組執行此操作。

dt = data.table(a = rep(c("a","b","c"), each = 5), b = 1:15) 
dt[, condition := b%%4 == 0] 
> dt 
    a b condition **desiredOutcome** 
1: a 1  FALSE   FALSE 
2: a 2  FALSE   FALSE 
3: a 3  FALSE   FALSE 
4: a 4  TRUE   TRUE 
5: a 5  FALSE   TRUE 
6: b 6  FALSE   FALSE 
7: b 7  FALSE   FALSE 
8: b 8  TRUE   TRUE 
9: b 9  FALSE   TRUE 
10: b 10  FALSE   TRUE 
11: c 11  FALSE   FALSE 
12: c 12  TRUE   TRUE 
13: c 13  FALSE   TRUE 
14: c 14  FALSE   TRUE 
15: c 15  FALSE   FALSE 

的desiredOutcome由a柱基於N=2和基團。得到這個的最好方法是什麼？

Answer 1

發現總結的shift的結果，即向量的列表的方式。

dt[, desiredOutcome := Reduce('+', shift(con, 0:2, fill = 0)), by = a]

Answer 2

我們可以嘗試

dt[dt[, {i1 <- which(condition); .I[i1:pmin((i1+2), .N)]} , a]$V1, 
    desiredOutcome:= TRUE][is.na(desiredOutcome), desiredOutcome := FALSE][] 
# a b condition desiredOutcome 
# 1: a 1  FALSE   FALSE 
# 2: a 2  FALSE   FALSE 
# 3: a 3  FALSE   FALSE 
# 4: a 4  TRUE   TRUE 
# 5: a 5  FALSE   TRUE 
# 6: b 6  FALSE   FALSE 
# 7: b 7  FALSE   FALSE 
# 8: b 8  TRUE   TRUE 
# 9: b 9  FALSE   TRUE 
#10: b 10  FALSE   TRUE 
#11: c 11  FALSE   FALSE 
#12: c 12  TRUE   TRUE 
#13: c 13  FALSE   TRUE 
#14: c 14  FALSE   TRUE 
#15: c 15  FALSE   FALSE