Question

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

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)) )

Answer 1

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]

，并提供：

    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

更改maxgap表示结转值的次数

1 个答案: