對3D陣列的每個第3維應用替換

Question

我有一個mxnxp（3D）數組A。我有一個2D矩陣B，其值在第一列中為1：m，在第二列中為1：n。 我想做的是在第三個維度（高度？）的每一個中對應於B給出的索引。 所以，

for (i in 1:p) {
    A[,,i][B] = NA
}

有沒有沒有for循環的方法？ 我在想類似

A_NA = apply(A,3,function(x) x[B] = NA)

但這是行不通的。

Answer 1

我們需要返回x ，然后將其分配回'A'

A[] <- apply(A,3,function(x) { x[B] = NA; x})

檢查OP的解決方案

for (i in 1:p) {
     A1[,,i][B] = NA
 }

identical(A, A1)
#[1] TRUE

數據

A <- array(1:40, c(5, 4, 2) )
B <- cbind(c(1, 2, 3, 4), c(2, 3, 1, 1))
p <- 2
A1 <- A