从 R 中的嵌套列表中提取向量

Question

我正在尝试根据同一个嵌套列表中另一个变量\元素的值从嵌套列表中提取一个向量。希望我的例子能解释我想要做什么。

首先，我有一个这样的列表：

## Create the inner lists
# Inner list 1
listInner1 <- list(
  value = c(0.25),
  index = c(1, 2, 3, 4, 5, 6, 7, 8, 9, 10),
  left  = c(),
  right = c()
)

listInner1$left$index  <- c(1, 2, 3, 4, 5)
listInner1$left$good   <- TRUE
listInner1$right$index <- c(6, 7, 8, 8, 10)
listInner1$right$good  <- TRUE

# Inner list 2
listInner2 <- list(
  value = c(1.5),
  index = c(1, 2, 3, 4, 5, 6, 7, 8, 9, 10),
  left  = c(),
  right = c()
)

listInner2$left$index  <- c(1, 2, 3)
listInner2$left$good   <- TRUE
listInner2$right$index <- c(4, 5, 6, 7, 8, 9, 10)
listInner2$right$good  <- TRUE

# Inner list 3
listInner3 <- list(
  value = c(0.5),
  index = c(1, 2, 3, 4, 5, 6, 7, 8, 9, 10),
  left  = c(),
  right = c()
)

listInner3$left$index  <- c(1, 2, 3, 4, 5)
listInner3$right$index <- c( 6, 7, 8, 9, 10)
listInner3$left$left$index  <- c(2, 4, 6, 8, 10)
listInner3$left$right$index <- c(1, 3, 5 ,7, 9)
listInner3$left$left$good <- TRUE
listInner3$left$right$good <- TRUE


# put all inner lists into single list object
listMiddle <- list(listInner1, listInner2, listInner3)

# one more list for fun
listMaster <- list(listMiddle)

如您所见，嵌套列表的某些left 和right 元素包含元素good = TRUE，而有些则不包含。

我想要做的是如果一个特定的嵌套列表包含元素 good = TRUE 然后从同一个嵌套列表中提取元素 index。

例如，为上述示例手动创建我想要的输出将如下所示：

ans <- list(
  index.1 = c(1, 2, 3, 4, 5),
  index.2 = c(6, 7, 8, 8, 10),
  index.3 = c(1, 2, 3),
  index.4 = c(4, 5, 6, 7, 8, 9, 10),
  index.5 = c(2, 4, 6, 8, 10),
  index.6 = c(1, 3, 5 ,7, 9)
)

对象ans 包含所有index 向量，这些向量包含在还包含good = TRUE 的嵌套列表中。

关于如何做到这一点的任何建议？

Answer 1

这是一个选项，我们使用rrapply 将嵌套元素bind 转换为更容易接近的格式，然后，我们获得“好”列的索引，从该位置索引中提取相应的“索引”元素在map2（基于TRUE 值）、transpose、list、keep 中循环仅大于 0 的元素 length、flatten、list 并设置名称（如果需要）

library(purrr)
library(rrapply)
library(stringr)
library(dplyr)
out <- rrapply(listMaster, how = 'bind')
i1 <- grep('good', names(out))
map2(out[i1-1], out[i1], `[`) %>%
   transpose %>%
   map( ~ keep(.x, lengths(.x) > 0)) %>%
   flatten %>%
   setNames(str_c('index.', seq_along(.)))

-输出

$index.1
[1] 1 2 3 4 5

$index.2
[1]  6  7  8  8 10

$index.3
[1] 1 2 3

$index.4
[1]  4  5  6  7  8  9 10

$index.5
[1]  2  4  6  8 10

$index.6
[1] 1 3 5 7 9