如何在R中向前/向后填充指定行数的最近观测值？

你可以通过自定义函数实现对非NA值的指定行数前后填充，支持向前、向后设置不同的填充行数，以下是两种可行方案：

方案一：Base R 实现（简洁高效） #

直接通过定位非NA值的位置，计算填充范围后批量赋值：

# 定义填充函数：n_forward=向前填充行数，n_backward=向后填充行数
fill_n_m <- function(x, n_forward, n_backward) {
  # 获取所有非NA值的位置和对应值
  non_na_pos <- which(!is.na(x))
  non_na_vals <- x[non_na_pos]
  
  # 初始化结果向量
  res <- rep(NA, length(x))
  
  # 遍历每个非NA值，填充指定范围
  for (i in seq_along(non_na_pos)) {
    pos <- non_na_pos[i]
    val <- non_na_vals[i]
    # 计算填充的起始和结束索引（避免超出向量边界）
    start_idx <- max(1, pos - n_forward)
    end_idx <- min(length(x), pos + n_backward)
    # 赋值填充
    res[start_idx:end_idx] <- val
  }
  
  return(res)
}

测试示例 #

使用你提供的向量测试：

# 原向量
vector <- c(NA, NA, NA, NA, NA, 1, NA, NA, NA, NA, 2, NA, NA, NA, NA, NA, NA, 3, NA, NA, NA, NA)

# 向前、向后各填充2行
filled_vector <- fill_n_m(vector, n_forward = 2, n_backward = 2)
filled_vector

输出结果：

[1] NA NA NA  1  1  1  1  1  2  2  2  2  2 NA NA  3  3  3  3  3 NA

如果需要设置不同的前后填充行数（比如向前1行、向后3行）：

filled_vector_diff <- fill_n_m(vector, n_forward = 1, n_backward = 3)
filled_vector_diff

输出结果：

[1] NA NA NA NA  1  1  1  1  1  2  2  2  2  2  2 NA  3  3  3  3  3 NA

方案二：dplyr 实现（适合数据框场景） #

如果你的数据是在数据框中，可以用dplyr的行号映射来实现：

library(dplyr)
library(tidyr)

# 将向量转为数据框
vector_df <- tibble(
  original_val = vector,
  row_num = row_number()
)

# 提取非NA值的行和对应值，生成填充范围映射
fill_mapping <- vector_df %>%
  filter(!is.na(original_val)) %>%
  rowwise() %>%
  mutate(
    # 定义填充的行范围（这里设置向前2行、向后2行）
    target_rows = list(max(1, row_num - 2):min(nrow(vector_df), row_num + 2))
  ) %>%
  unnest(target_rows) %>%
  select(target_rows, filled_val = original_val)

# 合并映射并填充值
filled_df <- vector_df %>%
  left_join(fill_mapping, by = c("row_num" = "target_rows")) %>%
  mutate(final_val = coalesce(filled_val, original_val))

# 提取填充后的向量
filled_vector_dplyr <- filled_df$final_val

说明 #

• 如果存在多个非NA值的填充范围重叠，后出现的非NA值会覆盖先出现的填充值，这符合常规的优先级逻辑。
• 两种方案都能灵活控制前后填充的行数，满足你的需求。

内容的提问来源于stack exchange，提问作者Molly Westbrook