想象一下，我有一点嵌套的 for 循环，它打印 a 和 b 的所有组合
a = c(1,2,3,4)
b = c(2,3,4,5)
for(i in a){
for(k in b){
print(i + k)
}}
所以输出看起来像这样
[1] 3
[1] 4
[1] 5
[1] 6
[1] 4
[1] 5
[1] 6
[1] 7
[1] 5
[1] 6
[1] 7
[1] 8
[1] 6
[1] 7
[1] 8
[1] 9
如何循环遍历这两个循环以获得只有 4 个项目的结果，即 a 和 b 中具有相同索引的元素的总和，类似于在 Python 中遍历字典？我想要这样的结果：
[1] 3
[1] 5
[1] 7
[1] 9
或这个
[1] 3 5 7 9
因此，我只需将 a 和 b 相加，就像将数据框中的两列相加以产生相同长度的三分之一。
我很感激任何帮助。
正如评论中指出的那样，在 中R，数学运算符+是Vectorized。这意味着默认情况下，您可以将向量作为参数提供给它们，它们将知道如何遍历每个输入向量中的元素。因此，简单地做a + b就会得到想要的结果。如果您真的想将其作为一个循环来执行，那么您可以不需要嵌套它 - 只需使用单个索引 ,i即可从两个输入向量中提取元素。此处可能有用的另一个选项是purrr::map2()将指定的函数应用于两个输入列表。
但是值得注意的是，如果您确实想查看所有成对组合，则可以使用该outer()功能。
# test vectors
a = c(1,2,3,4)
b = c(2,3,4,5)
# operate pairwise through the two vectors
a + b
#> [1] 3 5 7 9
# go through vectors as a loop
for(i in seq_along(a)){
print(a[i] + b[i])
}
#> [1] 3
#> [1] 5
#> [1] 7
#> [1] 9
# for more complex function can use purrr::map2 to run on two input lists
purrr::map2_dbl(.x = a, .y = b, `+`)
#> [1] 3 5 7 9
# operate on all combinations
outer(a, b, `+`)
#> [,1] [,2] [,3] [,4]
#> [1,] 3 4 5 6
#> [2,] 4 5 6 7
#> [3,] 5 6 7 8
#> [4,] 6 7 8 9
由reprex 包于 2022-04-13 创建(v2.0.1)
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句