繁体   English   中英

R对于每个第三维,通过向量乘以3D阵列的第二维

[英]R Multiply second dimension of 3D Array by a Vector for each of the 3rd dimension

当尝试将数组的第一个维度乘以向量的每个索引乘以第二个维度时,我的数组将转换为矩阵,并且事情变得松散。 我只能长时间做适当的乘法。

多么满口......

使用代码更容易解​​释......

Arr <- array(runif(10*5*3), dim = c(10,5,3))
dim(Arr)
Vect <- c(1:5)

Arr[,1,1] <- Arr[,1,1]*Vect[1]
Arr[,1,2] <- Arr[,1,2]*Vect[1]
Arr[,1,3] <- Arr[,1,3]*Vect[1]

Arr[,2,1] <- Arr[,2,1]*Vect[2]
Arr[,2,2] <- Arr[,2,2]*Vect[2]
Arr[,2,3] <- Arr[,2,3]*Vect[2]

Arr[,3,1] <- Arr[,3,1]*Vect[3]
Arr[,3,2] <- Arr[,3,2]*Vect[3]
Arr[,3,3] <- Arr[,3,3]*Vect[3]

Arr[,4,1] <- Arr[,4,1]*Vect[4]
Arr[,4,2] <- Arr[,4,2]*Vect[4]
Arr[,4,3] <- Arr[,4,3]*Vect[4]

Arr[,5,1] <- Arr[,5,1]*Vect[5]
Arr[,5,2] <- Arr[,5,2]*Vect[5]
Arr[,5,3] <- Arr[,5,3]*Vect[5]

如何将其清理为一个命令?

首先将Vect转换为数组,然后将元素乘以:

varr <- aperm(array(Vect, dim = c(5L, 10L, 3L)), perm = c(2L, 1L, 3L))

Arr <- varr * Arr

(当然,如果你想在一个命令中使用它,我们不需要存储varr

(另外,事实证明这基本上是sweep引擎盖下的......)

尝试:

sweep(Arr,2,Vect,FUN="*")

plyr包中的aaply()函数正是您正在寻找的。 它可以在任何维度的数组上运行,并根据需要拆分它们。 在这种情况下,您按行拆分:

library(plyr)
Arr2 <- aaply(Arr, 1, function(x,y){x*y}, Vect)

我们也可以复制'Vect'并乘以'Arr'。 col是一个方便的函数,它提供列的数字索引。

res1 <- Arr * Vect[col(Arr[,,1])]

或者我们明确地做rep

res2 <- Arr* rep(Vect, each=dim(Arr)[1])
identical(res1, res2)
#[1] TRUE

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM