r - 对 R 中的矩阵列表求和

Question

我有一个不同大小的矩阵列表。如何快速求和矩阵列表中的所有元素？

这是我当前的代码，但速度很慢。有没有更快的方法？

  for (i in 1: length(w)) {
    w_sum <- w_sum + sum(apply(w[[i]], 1:2, function (x) x^2))
  }

Answer 1

矩阵只是引擎盖下的向量，因此您可以unlist、平方和sum:

sum(unlist(w)^2)
#[1] 2393

使用@akrun 的示例数据，给出相同的结果。

如果你有大量的矩阵，它会快得多:

w <- rep(list(matrix(1:24,nrow=6)), 1e6)
system.time(sum(unlist(w)^2))
# user  system elapsed 
# 0.11    0.00    0.10 

system.time(sum(vapply(w, function(x) sum(x^2), numeric(1))))
# user  system elapsed 
# 2.17    0.00    2.17 

相比之下，在这种情况下循环也不必慢得离谱:

w_sum <- 0
system.time(for(i in seq_along(w)) { w_sum <- w_sum + sum(w[[i]]^2) } )
# user  system elapsed 
# 2.62    0.00    2.62