gpt4 book ai didi

r - 如何用 purrr 替换修改 for 循环

转载 作者:行者123 更新时间:2023-12-04 02:46:15 25 4
gpt4 key购买 nike

我想用 purrr 替代方案替换简单的 for 循环。我该如何编码并保持对象的原始结构

这是我的例子:

my_list <- list(
a = list(
list(
aaa = c(1:3),
aab = c(4:6),
aac = c(7:9)),
list(
aaa = c(10:12),
aab = c(13:15),
aac = c(16:18)),
list(
aaa = c(19:21),
aab = c(22:24),
aac = c(25:27))
))

和我原来的解决方案

x <- purrr::map_lgl(my_list$a, .f = ~ !is.null(.x$aaa))

my_list1 <- my_list
for (i in which(x)) {
my_list1$a[[i]]$aaa <- 99
}
str(my_list1)

但是,我也想用 purrr 解决方案替换我的 for 循环。我尝试过类似的方法,但 map 总是返回一个新列表而不是仅仅更新它

my_list2 <- map(c(1, 3),
function(x){
my_list[["a"]][[x]][["aab"]] <- 99
return(my_list)
}
)
str(my_list2)

Modify 应该更适用于这种情况,但我也无法让它发挥作用。

my_list3 <- purrr::map(
.x = which(x),
.f = function(i = .x) {
purrr::modify_in(.x = my_list,
.where = list("a", i, "aaa"),
.f = ~99)
}
)
str(my_list3)

新列表的结构应该与初始列表相同:

# str(my_list)
List of 1
$ a:List of 3
..$ :List of 3
.. ..$ aaa: int [1:3] 1 2 3
.. ..$ aab: int [1:3] 4 5 6
.. ..$ aac: int [1:3] 7 8 9
..$ :List of 2
.. ..$ aab: int [1:3] 13 14 15
.. ..$ aac: int [1:3] 16 17 18
..$ :List of 3
.. ..$ aaa: int [1:3] 19 20 21
.. ..$ aab: int [1:3] 22 23 24
.. ..$ aac: int [1:3] 25 26 27

# str(my_list1) is correct
List of 1
$ a:List of 3
..$ :List of 3
.. ..$ aaa: num 99
.. ..$ aab: int [1:3] 4 5 6
.. ..$ aac: int [1:3] 7 8 9
..$ :List of 2
.. ..$ aab: int [1:3] 13 14 15
.. ..$ aac: int [1:3] 16 17 18
..$ :List of 3
.. ..$ aaa: num 99
.. ..$ aab: int [1:3] 22 23 24
.. ..$ aac: int [1:3] 25 26 27

# str(my_list2) or str (my_list3) is not correct
List of 2
$ :List of 1
..$ a:List of 3
.. ..$ :List of 3
.. .. ..$ aaa: num 99
.. .. ..$ aab: int [1:3] 4 5 6
.. .. ..$ aac: int [1:3] 7 8 9
.. ..$ :List of 2
.. .. ..$ aab: int [1:3] 13 14 15
.. .. ..$ aac: int [1:3] 16 17 18
.. ..$ :List of 3
.. .. ..$ aaa: int [1:3] 19 20 21
.. .. ..$ aab: int [1:3] 22 23 24
.. .. ..$ aac: int [1:3] 25 26 27
$ :List of 1
..$ a:List of 3
.. ..$ :List of 3
.. .. ..$ aaa: int [1:3] 1 2 3
.. .. ..$ aab: int [1:3] 4 5 6
.. .. ..$ aac: int [1:3] 7 8 9
.. ..$ :List of 2
.. .. ..$ aab: int [1:3] 13 14 15
.. .. ..$ aac: int [1:3] 16 17 18
.. ..$ :List of 3
.. .. ..$ aaa: num 99
.. .. ..$ aab: int [1:3] 22 23 24
.. .. ..$ aac: int [1:3] 25 26 27

有什么提示我在使用 purrr 时出错了吗?

最佳答案

鉴于要修改的列表组件位于第二层嵌套,您也可以直接使用modify_depth访问它们:

library(purrr)

my_list <- modify_depth(my_list, 2, ~list_modify(.x, aaa = 99))

str(my_list)
#> List of 1
#> $ a:List of 3
#> ..$ :List of 3
#> .. ..$ aaa: num 99
#> .. ..$ aab: int [1:3] 4 5 6
#> .. ..$ aac: int [1:3] 7 8 9
#> ..$ :List of 3
#> .. ..$ aaa: num 99
#> .. ..$ aab: int [1:3] 13 14 15
#> .. ..$ aac: int [1:3] 16 17 18
#> ..$ :List of 3
#> .. ..$ aaa: num 99
#> .. ..$ aab: int [1:3] 22 23 24
#> .. ..$ aac: int [1:3] 25 26 27

注意:这也适用于第一个列表级别的多个列表组件(例如 $a$b ...),而无需添加额外的迭代步骤。


编辑:如modify_depth 文档中所列:

modify_depth(x, 2, fun) is equivalent to x <- modify(x, ~ modify(., fun))

为了仅替换第二个嵌套级别上的部分列表组件,我们可以以更详细的方式重写对 modify_depth 的调用:

library(purrr)

## replace aaa only for components 1 and 3
modify_loc <- c(TRUE, FALSE, TRUE)

my_list <- modify(my_list, ~imodify(., ~if(modify_loc[.y]) list_modify(.x, aaa = 99) else .x))

str(my_list)
#> List of 1
#> $ a:List of 3
#> ..$ :List of 3
#> .. ..$ aaa: num 99
#> .. ..$ aab: int [1:3] 4 5 6
#> .. ..$ aac: int [1:3] 7 8 9
#> ..$ :List of 3
#> .. ..$ aaa: int [1:3] 10 11 12
#> .. ..$ aab: int [1:3] 13 14 15
#> .. ..$ aac: int [1:3] 16 17 18
#> ..$ :List of 3
#> .. ..$ aaa: num 99
#> .. ..$ aab: int [1:3] 22 23 24
#> .. ..$ aac: int [1:3] 25 26 27

关于r - 如何用 purrr 替换修改 for 循环,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57159153/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com