如何在指定位置为R语言向量添加NA值？

已知初始向量：

x <- c(1,3,4,6,7)

以及用于指定NA插入位置的向量：

NAs <- c(2,5)

需要将NA插入到x的第2位和第5位，最终得到目标向量：

1 NA 3 4 NA 6 7

解决方案 #

方法1：索引定位填充法 #

先创建对应长度的空向量，再通过索引分别填充原向量元素和NA：

# 计算新向量的总长度
new_length <- length(x) + length(NAs)
# 初始化空向量
new_x <- vector(length = new_length)
# 确定原向量元素在新向量中的位置（排除NA插入位）
x_indices <- setdiff(1:new_length, NAs)
# 填充原向量元素
new_x[x_indices] <- x
# 在指定位置填充NA
new_x[NAs] <- NA

# 输出结果
new_x
# [1]  1 NA  3  4 NA  6  7

方法2：拆分拼接法 #

将原向量按插入位置拆分，再与NA交替拼接：

# 确保插入位置是升序排列（避免顺序错误）
NAs <- sort(NAs)
# 构造拆分节点，包含向量首尾
split_nodes <- c(0, NAs - 1, length(x))
# 初始化空向量用于存储结果
new_x <- c()

# 循环拆分并拼接
for (i in seq_len(length(split_nodes)-1)) {
  start <- split_nodes[i] + 1
  end <- split_nodes[i+1]
  # 若拆分片段有效，添加到结果中
  if (start <= end) {
    new_x <- c(new_x, x[start:end])
  }
  # 非最后一次循环时添加NA
  if (i < length(split_nodes)-1) {
    new_x <- c(new_x, NA)
  }
}

# 输出结果
new_x
# [1]  1 NA  3  4 NA  6  7

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