假设我有一个嵌套列表,其中包含一堆不同级别的数据框。我想提取出数据框的扁平列表。我该如何写这个使用purrr
功能?我应该看一下reduce
?
例如,给定数据:
s <- list(x = 1:10,
data = data.frame(report = LETTERS[1:5],
value = rnorm(5, 20, 5)),
report = list(A = data.frame(x = 1:3, y = c(2, 4, 6)),
B = data.frame(x = 1:3, y = c(3, 6, 9)),
z = 4:10,
other = data.frame(w = 3:5,
color = c("red", "green", "blue"))))
我希望函数返回:
list(data = data.frame(report = LETTERS[1:5],
value = rnorm(5, 20, 5)),
`report$A` = data.frame(x = 1:3, y = c(2, 4, 6)),
`report$B` = data.frame(x = 1:3, y = c(3, 6, 9)),
`report$other` = data.frame(w = 3:5,
color = c("red", "green", "blue")))
我写了一个递归函数:
recursive_keep <- function(.x, .f) {
loop <- function(.y) {
if(is.list(.y)) {
c(keep(.y, .f), flatten(map(discard(.y, .f), loop)))
} else if(.f(.y)) {
.y
} else {
NULL
}
}
loop(.x)
}
它可以被称为:
recursive_keep(s, is.data.frame)
它似乎适用于此示例,但它不保留名称信息。我希望保留足够的信息,以便我可以从原始对象中提取数据。也许这是一个更容易回答的问题?