如何在不使用for循环的情况下为数组指定维度元素加5（含NA处理）

需要在不使用for循环的前提下，为数组的指定维度子集（而非全部维度）的每个元素加5，最终得到维度与原数组完全一致的新数组。同时需处理大量NA值，要求NA + 5 = 5（即等效于na.rm = TRUE的处理逻辑）。

尝试使用sum和apply函数时遇到问题：要么得到单个元素总和，要么触发报错：

Error in apply(b[, , , 2], 5, FUN = sum, na.rm = T) : 
  'MARGIN' does not match dim(X)

原数组示例代码：

b <- array(seq(1,48,1), dim = c(4,2,3,2))
b[4,2,3,2] <- NA

尝试过的无效代码：

sum(b[,,,2], 5, na.rm = T)

期望输出：

(a <- array(c(seq(1,24,1),seq(25,47,1)+5, 5), dim = c(4,2,3,2)))

方法1：子集赋值 + replace处理NA #

直接定位目标维度子集，用replace将NA替换为0后再加5，这样NA会被转为0+5=5，符合需求。同时先复制原数组，避免修改原始数据：

# 复制原数组
result <- b
# 处理第4维的第2层
result[,,,2] <- replace(b[,,,2], is.na(b[,,,2]), 0) + 5

方法2：ifelse向量化判断 #

利用R的向量化特性，用ifelse直接对每个元素做判断：非NA元素加5，NA元素直接赋值为5：

result <- b
result[,,,2] <- ifelse(is.na(b[,,,2]), 5, b[,,,2] + 5)

方法3：先运算再替换NA #

先给目标子集整体加5，再将运算后产生的NA（原数组的NA）替换为5：

result <- b
temp <- b[,,,2] + 5
temp[is.na(temp)] <- 5
result[,,,2] <- temp

验证结果是否符合预期：

all.equal(result, a)
# 输出：TRUE

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