从n，k维的矩阵数组中减去n，k维的矩阵

Question

如果我有阵列A

A <- array(0, c(4, 3, 5))
for(i in 1:5) {
  set.seed(i)
  A[, , i] <- matrix(rnorm(12), 4, 3)
}

如果我有矩阵B

set.seed(6)
B <- matrix(rnorm(12), 4, 3)

从数组A的每个矩阵中减去B的代码是：

d<-array(0, c(4,3,5))
for(i in 1:5){
  d[,,i]<-A[,,i]-B
}

但是，使用“apply”系列中的函数执行相同计算的代码是什么？

Answer 1

这是sweep的用途。

sweep(A, 1:2, B)

Answer 2

也许不是很直观：

A[] <- apply(A, 3, `-`, B)

Answer 3

因为您在最后一个数组维度上循环，所以您可以执行以下操作：

d <- A - as.vector(B)

它会快得多。这与从矩阵中减去向量时的想法相同：向量被循环使用，因此将其减去每列。