r - 将嵌套的for循环存储到R中的单个向量中

Question

假设我有以下代码

b = 1:3
m = 5

for(j in 1:2){
  for(i in 1:5){
    print((1-i/m)* b[j] + (i/m)* b[j+1])
  }
}

如果我打印这个我得到以下输出

[1] 1.2
[1] 1.4
[1] 1.6
[1] 1.8
[1] 2
[1] 2.2
[1] 2.4
[1] 2.6
[1] 2.8
[1] 3

但是，现在我想将此数据存储到单个列向量中。当我用 print 代替空向量或列表 z[i] <- 这当然不起作用。有谁知道如何将 for 循环值转换为单个列向量？

Answer 1

我们可以启动一个向量并将输出附加到它

out <- c()
 for(j in 1:2){
  for(i in 1:5){
    out <- c(out, (1-i/m)* b[j] + (i/m)* b[j+1])
  }
}

df1 <- data.frame(out)

-输出

df1
   out
1  1.2
2  1.4
3  1.6
4  1.8
5  2.0
6  2.2
7  2.4
8  2.6
9  2.8
10 3.0

---

或者另一个选项outer来自base R

out <- c(t( outer(1:2, 1:5, FUN = function(j, i) (1-i/m)* b[j] + (i/m)* b[j+1])))

-输出

out
#[1] 1.2 1.4 1.6 1.8 2.0 2.2 2.4 2.6 2.8 3.0

---

或使用expand.grid( R 4.1.0)

expand.grid(j = 1:2, i = 1:5) |> 
      transform(out = (1-i/m)*b[j] + (i/m) * b[j + 1]) |> 
      subset(select = out)
   out
1  1.2
2  2.2
3  1.4
4  2.4
5  1.6
6  2.6
7  1.8
8  2.8
9  2.0
10 3.0

基准

j1 <- 1:200
i1 <- 1:500
# // outer 
system.time({
 out <- c(t( outer(j1, i1, FUN = function(j, i) (1-i/m)* b[j] + (i/m)* b[j+1])))

})
#  user  system elapsed 
#  0.004   0.000   0.004 

# // sapply

system.time({
out2 <- sapply(j1,function(j){
  sapply(i1,function(i){
    out <- (1-i/m)* b[j] + (i/m)* b[j+1]
    return(out)
  })
})
})
# user  system elapsed 
#  0.152   0.004   0.155 

Answer 2

碱基 R, sapply

b = 1:3
m = 5

df <- sapply(1:2,function(j){
  sapply(1:5,function(i){
    out <- (1-i/m)* b[j] + (i/m)* b[j+1]
    return(out)
  })
})

result <- c(df[c(1:5),c(1:2)]) 

结果

[1] 1.2 1.4 1.6 1.8 2.0 2.2 2.4 2.6 2.8 3.0

速度测试 - sapply 与 for 循环

b = 1:10001
m = 5

> system.time({ 
+ df <- sapply(1:10000,function(j){
+   sapply(1:5,function(i){
+     out <- (1-i/m)* b[j] + (i/m)* b[j+1]
+     return(out)
+   })
+ })
+ })
   user  system elapsed 
   0.19    0.00    0.19 
 
> system.time({ 
+ out <- c()
+ for(j in 1:10000){
+   for(i in 1:5){
+     out <- c(out, (1-i/m)* b[j] + (i/m)* b[j+1])
+   }
+ }
+ })
   user  system elapsed 
   3.00    0.02    3.02