gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-01 09:54:00 26 4
gpt4 key购买 nike

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

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

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

最佳答案

矩阵只是引擎盖下的向量,因此您可以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

关于r - 对 R 中的矩阵列表求和,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32389567/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com