用最小间隔插入 R 中的最后一个或下一个观察值的 NA

Question

我想在两个或多个间隙中使用上一个观察结果向前 (locf) 或下一个观察结果向后 (nocb) 来估算缺失值。

为了确定填充缺失值的方向（上/下），对数据框中的第一列（Interval）进行加权。插值（locf/nocb）应该从最小值结束（对应于向外的 NAs 行的间隔值）。默认的fillna是locf。

示例：异常 nafill 的逻辑：在 ID2 列第 5-7 行（包括标题）中，方向将为 nocb，因为 (Interval[7,] value= 50) 小于 Interval[5,] value= 100。

尝试使用 df1 <-df %>%fill(ID1, ID2, ID3,which.min(Interval)) %>% #default direction down fill(ID1, ID2, ID3, .direction = "up",which.min(Interval)) 无法正常工作，请协助更正它或建议另一种有效的 R 解决方案。预先感谢您的帮助。

输入

df = data.frame(
             Interval = c(0,20,80,100,50,50,130,100,70,80,200), 
             ID1 = c(0,1,1,0,NA,NA,NA,NA,1,NA,1), 
             ID2 = c(1,0,0,NA,NA,NA,1,1,3,NA,1), 
             ID3 = c(1,NA,1,0,3,NA,NA,NA,1,NA,1) 
)

Interval    ID1 ID2 ID3
0   0   1   1
20  1   0   NA
80  1   0   1
100 0   NA  0
50  NA  NA  3
50  NA  NA  NA
130 NA  1   NA
100 NA  1   NA
70  1   3   1
80  NA  NA  NA
200 1   1   1

预期输出

Interval    ID1 ID2 ID3
0   0   1   1
20  1   0   NA
80  1   0   1
100 0   1   0
50  0   1   3
50  0   1   3
130 0   1   3
100 0   1   3
70  1   3   1
80  NA  NA  NA
200 1   1   1

Answer 1

尚不清楚您的第一列是做什么用的。您的第二个预期列是locf 和nocb 的组合，但是您可以逐列使用标准nafill：

library(data.table)
apply(df, 2, function(j) {nafill(j, type = "locf")})

根据您的评论，如果您想让nafill依赖于df$Interval，您可以使用以下方法索引您的df：

StartIndex <- apply(df[,2:dim(df)[2]], 2, function(j) {
  which(diff(is.na(j)) == 1 ) + 1})
EndIndex <- apply(df[,2:dim(df)[2]], 2, function(j) {
  which(diff(is.na(j)) == -1) })

并指出df$Interval的开始和结束是增加还是减少：

lapply(1:(dim(df)[2]-1), function(i) {
  lapply(1:length(StartIndex[[i]]), function(j) {
    range <- c(StartIndex[[i]][j], EndIndex[[i]][j])
    if (range[1] != range[2]) {
      df$Interval[range[1]] - df$Interval[range[2]]
    }
  })
})