如何在R语言中从嵌套for循环中输出结果为向量?
R语言嵌套for循环结果转为向量的实现方法
方法1:修改原有for循环,改动最小
提前预分配固定长度的向量存储每一步的计算结果,避免动态拼接的性能损耗:
# 计算总元素个数,预分配空向量 total_len <- length(0:3) * length(1:3) res <- numeric(total_len) count <- 1 for(i in 0:3){ for(j in 1:3){ res[count] <- 0.03*(i-1)+exp(-j) count <- count + 1 } } # 直接输出向量结果 res
方法2:纯向量化操作(更符合R语言使用习惯,效率更高)
不需要写循环,直接生成所有i、j的组合批量计算:
# 生成i和j的所有配对组合,顺序和原循环完全匹配 param_grid <- expand.grid(i = 0:3, j = 1:3) res <- 0.03*(param_grid$i - 1) + exp(-param_grid$j)
也可以用outer函数实现一行计算:
res <- as.vector(outer(0:3, 1:3, function(i,j) 0.03*(i-1) + exp(-j)))
以上所有方法输出的向量数值、顺序均与原嵌套循环逐次print的结果完全一致
内容的提问来源于stack exchange,提问作者yoyo
相关产品推荐
相关产品推荐

