我应该如何在 R 中对这个简单的 for 循环进行矢量化？

Question

要矢量化的循环：

n = 1000000  
x = numeric(n)  
for (I in 1:n) x[i] = rpois(1, 3) + rpois(1, 5)  

原版太慢了，我对 for 循环或 apply 系列函数还没有太多经验，所以我想我只需要看一个例子来说明如何做到这一点，我就明白了。谢谢

Answer 1

试试

x = rpois(n, 3) + rpois(n, 5)

rpois(n,3) 已经创建了许多平局，因此无需循环。

要查看此内容，请注意以下几点：

循环创建：

set.seed(42)
n <- 10
x = numeric(n)

for (i in 1:n){
  x[i] <- rpois(1, 3) 
}

一次创建：

set.seed(42)
y <- rpois(n, 3) 

检查结果是否相等：

all.equal(x,y)
[1] TRUE

请注意，加法的结果会略有不同，因为随机数生成器中的增量会有所不同，但这是一个小细节。关键是不需要循环。