Суммировать список матриц в 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 

Loops также не должны быть невероятно медленно в этом обстоятельстве в сравнении:

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