是否可以使用 map() 从 R 中的列表中提取多个项目?

Is it possible to extract multiple items from a list in R with map()?

通过使用 purrr 包中的各种 map() 函数,您可以将整数或字符串传递给函数并从子列表中提取元素。这些工作正如我所期望的那样:

listy <- list(A = list("silence", "cats", X = list(Aa = "dogs", "birds")),
              B = list("silence", "fish-head", Y = list(Bb = "fish-face", "squirrel")))

> str(listy)
List of 2
 $ A:List of 3
  ..$  : chr "silence"
  ..$  : chr "cats"
  ..$ X:List of 2
  .. ..$ Aa: chr "dogs"
  .. ..$   : chr "birds"
 $ B:List of 3
  ..$  : chr "silence"
  ..$  : chr "fish-head"
  ..$ Y:List of 2
  .. ..$ Bb: chr "fish-face"
  .. ..$   : chr "squirrel"

list1 <- listy %>% map(1) %>% str  # returns "silence" from A and B
list2 <- listy %>% map(2) %>% str  # returns "cats" and "fish-head"
list3 <- listy %>% map(c(3, 1)) %>% str # returns "dogs" and "fish-face" from the lists X and Y within the list.

我的问题是,如何从此列表中提取多个元素?如果我想要 A 和 B 的 "silence",A 和 B 的 "cats" 和 "fish-head"(换句话说,A 和 B 的元素 1 和 2),是否可以使用 map()?如果没有,最好的方法是什么?

以下是我认为可行的方法:

list4 <- listy %>% map(1, 2) %>% str

其中 1 表示每个子列表中的第一个元素,2 表示第二个。但是这个returns和上面的list1是一样的。使用 c(1, 2) 不起作用,因为它指的是嵌套结构(即 [[1]][[2]])。我查看了文档和通过 Google 找到的一些示例,但没有运气。有什么想法吗?

更新:我应该解释一下,理想情况下我希望能够按名称 select 元素,如 "silence" 中那样。然而,这似乎不太奏效。 (我有几个大列表,其中我想要更改的元素的位置)

是这样的吗?

library(purrr)
listy %>% map(., function(x) c(x[[1]], x[[2]]))

$A
[1] "silence" "cats"   

$B
[1] "silence"   "fish-head"

data.frame

的形式得到输出
listy %>% map_df(., function(x) c(x[[1]], x[[2]]))

# A tibble: 2 x 2
        A         B
    <chr>     <chr>
1 silence   silence
2    cats fish-head

或者,按照@Richard Scriven 的建议,

map(listy, ~ unlist(.[1:2]))

这里有两个选项 lapply

要保留嵌套列表结构,请使用

lapply(listy, "[", 1:2)
$A
$A[[1]]
[1] "silence"

$A[[2]]
[1] "cats"


$B
$B[[1]]
[1] "silence"

$B[[2]]
[1] "fish-head"

要return 单个向量列表使用

lapply(listy, function(i) unlist(i[1:2]))
$A

"silence"    "cats" 

$B

  "silence" "fish-head" 

您可以将 data.frame 中的第二种方法包装到 return 一个 data.frame.

data.frame(lapply(listy, function(i) unlist(i[1:2])))
        A         B
1 silence   silence
2    cats fish-head