Пометка как 1 n последовательных строк при выполнении условия

Question

Предположим, у меня есть список, который содержит 0 или 1.

Рядом с этим (назовите его ColA), я хочу добавить столбец (назовите его ColB), который принимает следующее значения:

i. если ColA == 1, то следующие n строки в ColB должны быть помечены как 1

ii. если ColA == 0, то соответствующая строка в ColB должна быть помечена 0, если только она не является одной из этих n строк, следующих за 1 в ColA

Может кто-нибудь предложить эскиз код / ​​идея как этого добиться?

Любое решение в R, SQL, Python или C ++ приветствуется.

Answer 1

Ответ в R:

for (i in (1:nrow(db))){
    if (db$ColA[i] != 0){
        for (j in (i:(n+i-1))){
            db$ColB[j] <- 1
        }
    } 
}