Question

假设我有两个矩阵x和y，两者的尺寸均为100x2。我想创建一个列表，对于x和y的每一行，我有矩阵t（x）％*％y。例如，通过for循环：

x = matrix(rnorm(10), nrow = 5)
y = matrix(rnorm(10), nrow = 5)
myList = list()
for(i in 1:5){
    myList[[i]] = t(x[i, , drop = FALSE]) %*% y[i, ]
}

有更有效的方法来进行此计算吗？我试图找出如何表达矩阵乘法，但没有运气。我也考虑了mapply，但似乎我需要将x和y转换为向量列表而不是矩阵才能使用mapply，我怀疑这是正确的方法。

Answer 1

使用Map

的一种方法

Map(function(x,y) matrix(x,ncol=1)%*%y ,
               split(x, row(x)), split(y, row(y)))

Answer 2

您可以使用

缩短（并可能略微加快）您的代码

NewList <- list()
for (i in 1:nrow(x)) NewList[[i]] <- outer(x[i,],y[i,])
#> all.equal(NewList,myList)
#[1] TRUE

或等同地

for (i in 1:nrow(x)) NewList[[i]] <- x[i,] %o% y[i,]

Answer 3

Map似乎是最好的方法：

library(rbenchmark)

x = matrix(rnorm(10000), nrow = 5000)
y = matrix(rnorm(10000), nrow = 5000)
myList = list()

loopTest = function(){
    for(i in 1:nrow(x)){
        myList[[i]] = t(x[i, , drop = FALSE]) %*% y[i, ]
    }
}

loopTest2 = function(){
    for(i in 1:nrow(x)){
        myList[[i]] = outer(x[i, ], y[i, ])
    }
}

mapTest = function(){
    Map(function(x,y) matrix(x,ncol=1)%*%y ,
                   split(x, row(x)), split(y, row(y))) 
}

mapplyTest = function(){
    mapply(function(x,y) matrix(x,ncol=1)%*%y,
           x = split(x, row(x)), y = split(y, row(y))) 
}

benchmark(loopTest(), mapTest(), mapplyTest(), replications = 100)

这给了我：

test        elapsed
loopTest()   10.471
loopTest2()  12.225
mapplyTest()  3.100
mapTest()     2.252

然而，循环方法确实在较小的数据集上获胜，比如只有5行。

“矢量化”矩阵乘法

3 个答案: