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参数，您可以选择如何处理剩余的后续NAs。你可以选择"rev“(反转)，"mean”或"keep“。选择"rev“会在开始时对locf无法填充的所有NAs执行nocb。如果您首先选择option = "nocb“而不是"locf”，那么它当然会在后面的NAs中填充"locf“。