在多个列表中应用操作

时间:2017-07-20 17:02:02

标签: r list matrix

我知道如果我有一个矩阵列表,我可以使用Reduce函数在所有矩阵中应用一个操作。例如:

l <- list(matrix(rnorm(16), 4, 4), matrix(rnorm(16), 4, 4))
Reduce(`*`, l)

但是,如果我想在多个列表中应用此操作,该怎么办?我可以使用for循环进行暴力破解,但我觉得应该有更好的方法。我可以使用mapply

执行两个列表
l2 <- l
mapply(`*`, l, l2, SIMPLIFY = FALSE)

但如果我有两个以上,我不知道如何解决这个问题。

以下想法都会导致错误:

l3 <- l2
mapply(`*`, l, l2, l3, SIMPLIFY = FALSE)
Error in .Primitive("*")(dots[[1L]][[1L]], dots[[2L]][[1L]], dots[[3L]][[1L]]) :
  operator needs one or two arguments

Reduce(`*`, list(l, l2, l3))
Error in f(init, x[[i]]) : non-numeric argument to binary operator

所需的输出是长度为2的列表,其中每个列表中的每个矩阵的元素乘积。蛮力循环看起来像这样:

out <- vector("list", length = 2)
for(i in 1:2){
  out[[i]] <- l[[i]] * l2[[i]] * l3[[i]]
}

1 个答案:

答案 0 :(得分:6)

ReduceMap的这种组合将在基数R中产生所需的结果。

# copy the matrix list
l3 <- l2 <- l

out2 <- Reduce(function(x, y) Map(`*`, x, y), list(l, l2, l3))

返回

out2
[[1]]
              [,1]        [,2]        [,3]       [,4]
[1,] -5.614351e-01 -0.06809906 -0.16847839  0.8450600
[2,] -1.201886e-05  0.02008037  5.64656727 -2.4845526
[3,]  5.587296e-02 -0.54793853  0.02254552  0.4608697
[4,] -9.732049e-04 11.73020448  1.83408770 -1.4844601

[[2]]
              [,1]         [,2]       [,3]        [,4]
[1,] -4.7372339865 -0.398501528  0.8918474  0.12433983
[2,]  0.0007413892  0.151864126 -0.2138688 -0.10223482
[3,] -0.0790846342 -0.413330364  2.0640126 -0.01549591
[4,] -0.1888032661 -0.003773035 -0.9246891 -2.30731237

我们可以检查这与OP中的for循环相同。

identical(out, out2)
[1] TRUE