如何在R中对3D数组与矩阵执行运算并直接返回3D数组？

方法1：无循环向量化运算（推荐） #

这是效率最高的方案，通过扩展b的维度使其与a的结构匹配，利用R的广播机制直接调用myfunc完成运算，结果直接是3D数组：

# 定义自定义函数
myfunc <- function(x, y) { x * y }

# 扩展b的维度：从2x3转为2x1x3，实现和a的2x2x3维度广播匹配
b_expanded <- array(b, dim = c(nrow(b), 1, ncol(b)))

# 直接运算，自动广播维度
result <- myfunc(a, b_expanded)

验证：a[,,1]是全1的2x2矩阵，b[,1]是c(0, 0.2)，运算后result[,,1]会是每行对应乘b[,1]的结果，完全符合需求，且无需额外拼接步骤。

方法2：purrr列表遍历+数组重塑 #

如果你习惯遍历式写法，用purrr的映射函数更直观，最后直接将结果列表重塑为3D数组：

library(purrr)

# 遍历第三维度的每个索引，配对a的切片和b的对应列
result_list <- map(1:dim(a)[3], ~myfunc(a[,,.x], b[,.x]))

# 将列表展开并按原数组维度重塑
result <- array(unlist(result_list), dim = dim(a))

这种写法逻辑清晰，相比sapply+abind，不需要额外的包来拼接数组，直接用基础R的array即可完成重塑。

方法3：apply遍历第三维度 #

利用基础R的apply按第三维度遍历，再调整维度得到目标数组：

# 按a的第三维度（MARGIN=3）遍历，传入对应列的b值
result_temp <- apply(a, MARGIN = 3, function(x, b_col) myfunc(x, b_col), b_col = b)

# 将临时结果转为数组并调整维度顺序
result <- aperm(array(result_temp, dim = c(dim(a)[1:2], dim(a)[3])), c(1,2,3))

这里apply返回的结果是按列排列的矩阵，需要先转为数组再用aperm调整维度顺序，最终得到正确的2x2x3结构。

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