更改值结转次数的 maxgap

我有一个类似于以下内容的数据框：

library(data.table)
test <- data.table(data.frame("value" = c(5,NA,8,NA,NA,8,6,NA,NA,10),
                              "locf_N" = c(1,NA,1,NA,NA,1,2,NA,NA,2)) )

在这个数据框中，我有一个变量，指示我可以结转最后一次观察的时间（locf_N）。这并不是所有观察结果的固定数字。我尝试在 na.locf 函数中使用 maxgap 参数来实现此目的，但它实际上并不是我正在寻找的。

require(zoo)
test[,value := na.locf(value, na.rm = FALSE, maxgap = 1)]
test[,value := na.locf(value, na.rm = FALSE, maxgap = locf_N)]

是否有任何参数可以设置最后一次观察可以结转的次数？欢迎任何想法。

期望的输出：

output <- data.table(data.frame("value" = c(5,5,8,8,NA,8,6,6,6,10),
                                "locf_N" = c(1,NA,1,NA,NA,1,2,NA,NA,2)) )

cumsum(!is.na(value))是一个分组向量，它将每个非 NA 与以下 NA 分组。然后，对于每个这样的组，重复第一个值所需的次数，并将其余值保留为 NA。

test[, list(value = replace(value, 1:min(.N, locf_N[1] + 1), value[1]), locf_N), 
        by = cumsum(!is.na(value))][, -1]

giving:

    value locf_N
 1:     5      1
 2:     5     NA
 3:     8      1
 4:     8     NA
 5:    NA     NA
 6:     8      1
 7:     6      2
 8:     6     NA
 9:     6     NA
10:    10      2