R中为组的每个成员分配值的快速方法

Question

我需要为组中的每个成员分配一个权重。 我已经创建了以下循环来创建weights_ind 向量（单个权重的向量），但是我的循环太慢了（2000 个人需要 2 秒）：

# Data example
   n.group = 10
   n.tot = 1000
   groups = sample(1:10, n.tot, replace = TRUE)
   weights_by_group = rexp(n.group, rate = 1)


   weights_ind = rep(NA, n.tot)
     for(i in 1:n.tot){
     for(j in 1:n.group) {
     if(groups[i] == j) 
       weights_ind[i] = weights_by_group[j]
     }}

在我的情况下，我可以使用哪个更快的功能？

Answer 1

由于组从 1 开始编号，因此您可以使用 groups 向量来索引 weights_by_group向量。

weights_ind = weights_by_group[groups]

在我的电脑上，对于 2000 个人来说，这大约需要 10 微秒。

如果您的组不是数字，您可以使用组标识符命名weights_by_group 向量，并仍然使用groups 来索引：

groups = sample(letters[1:n.group], n.tot, replace = TRUE)
weights_by_group = rexp(n.group, rate = 1)
names(weights_by_group) = letters[1:n.group] 

weights_ind = weights_by_group[groups]