使用线性插值法在一个变量中插值 NA 值

Question

尝试在 dataframe 中插入 NA 值，并尝试使用zoo::na.approx但这只会返回等间距的值。

为了演示，给定 dataframe：

library(dplyr)
library(zoo) 
df <- data.frame(
      a = c(1,5 ,6 , 10),
      b = c(100, NA, NA, 1000)
)

试过这个：

df %>% 
  mutate(b = zoo::na.approx(b, rule = 2))

而是想得到这个：

desired_df <- data.frame(
  a = c(1, 5 ,6 , 10),
  b = c(100, 500, 600, 1000)
)

是否有现成的 function 来执行此操作，或者必须使用线性插值公式编写：

Answer 1

lin_interp <- function(known, unknown) {
  # unknown object
  i1 <- !is.na(unknown)

  approx(x = known[i1], 
         y = unknown[i1], 
         xout = known)$y
}

提高代码可读性

output 如所愿：

df %>% 
  mutate(b = lin_interp(a, b))

Answer 2

实际上，它在zoo中可用：

desired_df <- as.zoo(df, order.by = df$a) %>% 
  na.approx()

如您所见，您需要将数据框转换为zoo class。

Answer 3

代码na.approx不起作用的原因是它通过检查 class 来调度方法。 如果我们检查

library(zoo)
methods('na.approx')
#[1] na.approx.default na.approx.ts*     na.approx.zoo*    na.approx.zooreg*

na.approx.default方法是应用于 OP 数据的方法，因为它不是zoo object。 如果我们检查源代码，这些方法的行为是不同的

na.approx.default

和*函数

getAnywhere('na.approx.zoo')

因此，正如提到的另一个答案，如果我们转换为适当的 class，生成的 output 也会有所不同，因为调度的方法是基于class