在R语言中计算n维数组中每个矩阵的均值

嘿，作为R语言新手碰到数组操作的困惑太正常啦！我来帮你搞定计算n维数组里每个矩阵均值的事儿～

核心方法：用apply()函数精准操作维度 #

R里的apply()函数专门用来对数组/矩阵的指定维度应用函数，刚好能解决你的问题。关键是要选对MARGIN（操作维度）：

• 假设你的n维数组中，每个矩阵是前2个维度（行×列），后面的维度是这些矩阵的分组/索引（比如第3、4...n维）
• 你需要把MARGIN参数设为后面这些维度的编号（比如3，或者c(3,4)这样的组合），让apply()遍历每个对应的矩阵并计算均值

示例1：3维数组（最常见的多矩阵场景） #

先创建一个3维数组，包含3个2×2的矩阵：

# 生成3维数组：2行、2列、3个矩阵
arr_3d <- array(1:12, dim = c(2, 2, 3))
print(arr_3d)

现在计算每个矩阵的均值：

# 针对第3个维度（每个独立矩阵）计算均值
matrix_means <- apply(arr_3d, MARGIN = 3, FUN = mean)
print(matrix_means)
# 输出结果：[1]  3.5  7.5 11.5

解释：MARGIN=3表示我们要沿着第3个维度遍历，每次取出一个2×2的矩阵，用mean()计算它的均值，最终得到一个对应每个矩阵均值的向量。

示例2：更高维的n维数组 #

如果是4维数组（比如2×2的矩阵，按「组」和「批次」分成2×2共4个矩阵）：

# 生成4维数组：2行、2列、2组、2批次
arr_4d <- array(1:32, dim = c(2,2,2,2))
# 遍历第3、4维度的所有组合，计算对应矩阵的均值
matrix_means_4d <- apply(arr_4d, MARGIN = c(3,4), FUN = mean)
print(matrix_means_4d)
# 输出结果是一个2×2的数组：
#      [,1] [,2]
# [1,]  4.5 20.5
# [2,] 12.5 28.5

这里MARGIN=c(3,4)告诉apply()要同时遍历第3和第4维度的每一个组合，每个组合对应一个2×2的矩阵，计算后的结果维度和你指定的MARGIN维度一致。

新手常踩的坑 #

• 直接用mean(arr)：这会计算整个数组所有元素的均值，而不是每个矩阵的
• 选错MARGIN：比如设成1或2，那会计算每行/每列的均值，不是你要的矩阵整体均值

如果你的数组结构和示例不太一样，随时调整MARGIN的参数就行～

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