locf 和 nocb 函数

Question

我想以此为基础创建一个函数：

locf <- function(x) {
  a <- x[1]
  for (i in 2:length(x)) {
    if (is.na(x[i])) x[i] <- a
    else a <- x[i]
  }
  return(x)
}

这个函数只是 LOCF，我想创建一个函数，它是所有缺失数据的 LOCF 和第一个缺失观测值的 NOCB。

我的数据：

data<-c(NA, 24.107, NA, 26.912, NA, 30.193, NA, 19.003, NA, NA, 28.578, NA, NA, 33.484, 32.952, 13.574, NA, NA, 38.782, NA, NA, NA, 28.804, NA, 27.042, NA, NA, NA, NA, 21.344, NA, 21.507, NA, NA, NA, 16.508, NA, NA, 14.015, 11.738, 15.055)

我想得到什么：

locf<-c(24.107, 24.107, 24.107, 26.912, 26.912, 30.193, 30.193, 19.003, 19.003, 19.003, 28.578, 28.578, 28.578, 33.484, 32.952, 13.574, 13.574, 13.574, 38.782, 38.782, 38.782, 38.782, 28.804, 28.804, 27.042, 27.042, 27.042, 27.042, 27.042, 21.344, 21.344, 21.507, 21.507, 21.507, 21.507, 16.508, 16.508, 16.508, 14.015, 11.738, 15.055)

提前谢谢你。

Answer 1

如果您通过“第一次缺失的观察”来引用列表的第一个元素，则可以使用以下方法：

my_function <- function(my_list) {
  for(i in 1:length(my_list)) {
    if(is.na(my_list[i])) {
      if(i == 1) {
        my_list[i] <- my_list[i + 1]
      } else {
        print(i)
        my_list[i] <- my_list[i - 1]
      }
    }
  }
  return(my_list)
}

如果您想以不同的方式处理列表的第一个 NA 观察值，而不管其位置如何（即它可能是整个列表的第三个元素），您可以调整函数：

my_function <- function(my_list) {
  first <- 0
  for(i in 1:length(my_list)) {
    if(is.na(my_list[i])) {
      if(first == 0) {
        my_list[i] <- my_list[i + 1]
        first <- 1
      } else {
        print(i)
        my_list[i] <- my_list[i - 1]
      }
    }
  }
  return(my_list)
}

Answer 2

这会将第一个 na 设置为下一个值，然后通过它们的 locf 值重置后续 na。

#library(zoo)
Nocb1.locf <- function (x){ wh1st <- which(is.na(x))[1]; 
   x[wh1st] <- x[wh1st + 1]
   x <- zoo::na.locf(x) }

Answer 3

imputeTS 包中正好有这个函数：

library(imputeTS)
imputeTS::na.locf(x, option = "locf", na.remaining = "rev")

使用 na.remaining 参数，您可以选择如何处理剩余的尾随 NA。您可以在“rev”（反向）、“mean”或“keep”之间进行选择。选择“rev”对所有不能被 locf 填充的 NA 执行 nocb。如果您首先选择选项 =“nocb”而不是“locf”，那么它当然会用“locf”填充尾随的 NA。