Как пометить, когда скользящие соединения вступают в силу в data.table

Повторите вектор 1,0,0 до длины .N для каждой группы, используя rleid для формирования групп.

k <- 3 # restart after this number of events that are the same
one.zeros <- rep(1:0, c(1, k-1))  # length k vector.  For k=3, c(1,0,0)
DT[, V3 := rep(one.zeros, length = .N), by = rleid(V2)]

давая:

> DT
   V1 V2 V3
1:  1  a  1
2:  2  a  0
3:  3  a  0
4:  4  a  1
5:  5  a  0
6:  6  a  0
7:  7  c  1
8:  8  c  0
9:  9  d  1

Примечание

Вход DT:

library(data.table)
DT <- data.table(1:9, c("a", "a", "a", "a", "a", "a", "c", "c", "d"))

Вы можете проверить, совпали ли ваши ключи (V1 в вашем примере) во время присоединения:

a1[b, .(V2 = x.V2, event = isTRUE(x.V1 == i.V1)), roll = 2, by = .EACHI]
   V1 V2 event
1:  1  a  TRUE
2:  2  a FALSE
3:  3  a FALSE
4:  4  a  TRUE
5:  5  a FALSE
6:  6  a FALSE
7:  7  c  TRUE
8:  8  d  TRUE

a2[b, .(V2 = x.V2, event = isTRUE(x.V1 == i.V1)), roll = 2, by = .EACHI]
   V1   V2 event
1:  1    a  TRUE
2:  2    a FALSE
3:  3    a FALSE
4:  4 <NA> FALSE
5:  5 <NA> FALSE
6:  6 <NA> FALSE
7:  7    c  TRUE
8:  8    d  TRUE

Принимая во внимание комментарии в этом ответе:

set.seed(5438L)
n <- 1e5
a <- data.table(
        sample(2 * n, n, replace = FALSE),
        sample(c("a", "b", "c"), n, replace = TRUE),
        key = "V1"
)

b <- data.table(1:(2 * n), key = "V1")

library(microbenchmark)

microbenchmark(
        during = a[b, c(.SD, event = isTRUE(x.V1 == i.V1)), roll = 2, by = .EACHI],
        after = a[b, roll = 2][a, event := !is.na(i.V1), on = "V1"],
        times = 30L
)
Unit: milliseconds
   expr       min        lq      mean    median        uq       max neval cld
 during 767.49338 771.49878 795.02283 776.20243 787.96382 964.11575    30   b
  after  26.14068  26.46543  28.58425  27.51831  29.73552  37.36052    30  a 

Так что ответ IceCreamToucan, вероятно, лучше в этом случае.

Другой вариант ответа @GaborGrothendieck:

DT[, v := 0L]
DT[(rowid(rleid(V2)) - 1L) %% 3 == 0, v := 1L][]

   V1 V2 v
1:  1  a 1
2:  2  a 0
3:  3  a 0
4:  4  a 1
5:  5  a 0
6:  6  a 0
7:  7  c 1
8:  8  c 0
9:  9  d 1

Это просто выполняет арифметические действия над rowid (1,2,3...) внутри каждой rleid группы.

Я не уверен, что полностью понял намерения ОП, стоящие за вопросом, но я считаю, что это можно решить альтернативно без.

Если я правильно понимаю, цель - расширить список событий

   V1 V2
1:  1  a
2:  4  a
3:  7  c
4:  8  d

в период серия активных событий

   V1 V2 event
1:  1  a     1
2:  2  a     0
3:  3  a     0
4:  4  a     1
5:  5  a     0
6:  6  a     0
7:  7  c     1
8:  8  d     1

Во входной таблице data.table V1 обозначает индекс времени, когда произошло событие типа V2. Итак, мы можем повторять каждое значение V2 до следующего события. Столбец event создается путем сравнения позиций, указанных в a1$V1, с последовательностью индексов времени:

a1[, .(1:max(V1), rep(V2, c(diff(V1), 1)), event = +(1:max(V1) %in% V1))]

   V1 V2 event
1:  1  a     1
2:  2  a     0
3:  3  a     0
4:  4  a     1
5:  5  a     0
6:  6  a     0
7:  7  c     1
8:  8  d     1

(Обратите внимание, что 1:max(V1) %in% V1 имеет логический тип, +(1:max(V1) %in% V1) приводит его к целому числу.)

Кроме того, есть два варианта, которые возвращают один и тот же результат, но могут отличаться по производительности (не тестировались):

a1[, .(1:max(V1), rep(V2, c(diff(V1), 1)), event = replace(rep(0L, max(V1)), V1, 1L))]

использует replace() для создания столбца event.

a1[, .(1:max(V1), rep(V2, c(diff(V1), 1)), event = 0L)][a1$V1, event := 1L][]

инициализирует столбец event, но выборочно обновляет его на следующем шаге.