如何用dplyr按顺序以向量元素替换数据框某列的NA值？

问题背景 #

现有如下演示数据：

set.seed(123)
col1 <- runif(30)
col1[sample(length(col1), 20)] <- NA
col2 <- seq_len(30)
df <- data.frame(col1, col2)

数据框df的col1列存在20个NA值，需要用向量vec <- c("A1", "B1", "C1", "D1", "E1")按顺序循环替换这些NA值（即遵循A1→B1→C1→D1→E1→A1→B1...的顺序填充所有NA）。

尝试过以下两段代码，但未达到预期效果：

# 方法1
df2 <- df %>%
  mutate(col1 = ifelse(is.na(col1), vec, col1))

# 方法2
df2$col1 <- replace(df2$col1, is.na(df2$col1), vec)

解决方案 #

问题核心是直接使用vec替换时，R的向量回收机制没有按NA出现的顺序逐个循环赋值。以下是两种符合需求的dplyr实现方式：

方法1：利用累积计数生成循环索引 #

通过标记NA的出现顺序，生成对应循环的向量元素进行替换：

library(dplyr)

vec <- c("A1", "B1", "C1", "D1", "E1")

df2 <- df %>%
  mutate(
    # 为每一个NA行生成递增计数
    na_seq = cumsum(is.na(col1)),
    col1 = ifelse(
      is.na(col1),
      # 按计数循环提取vec中的元素
      vec[(na_seq - 1) %% length(vec) + 1],
      as.character(col1)  # 转换为字符型，避免类型冲突
    )
  ) %>%
  select(-na_seq)  # 移除临时计数列

方法2：预先生成对应长度的循环填充向量 #

先统计NA数量，生成与NA数量匹配的循环填充向量，再执行替换：

library(dplyr)

vec <- c("A1", "B1", "C1", "D1", "E1")
na_total <- sum(is.na(df$col1))
# 生成刚好覆盖所有NA的循环向量
fill_vec <- rep(vec, length.out = na_total)

df2 <- df %>%
  mutate(
    col1 = replace(
      as.character(col1),
      is.na(col1),
      fill_vec
    )
  )

说明：原col1为数值型，替换为字符型填充值后，列类型会变为字符型，若需保持类型一致性，可根据实际需求调整（例如将原数值转换为字符型）。

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