R:重新列出清单

时间:2014-04-14 09:08:17

标签: r list

This question有一个很好的解扁平列表解决方案,同时保留了他们的数据类型(unlist没有):

flatten = function(x, unlist.vectors=F) {
    while(any(vapply(x, is.list, logical(1)))) {
        if (! unlist.vectors)
            x = lapply(x, function(x) if(is.list(x)) x else list(x))
        x = unlist(x, recursive=F)
    }
    x
}

如果我给它以下列表,它的行为符合预期:

> a = list(c(1,2,3), list(52, 561), "a")
> flatten(a)
[[1]]
[1] 1 2 3

[[2]]
[1] 52

[[3]]
[1] 561

[[4]]
[1] "a"

现在我想像a那样重组平面列表。 relist悲惨地失败了:

> relist(flatten(a), skeleton=a)
[[1]]
[[1]][[1]]
[1] 1 2 3

[[1]][[2]]
[1] 52

[[1]][[3]]
[1] 561

[[2]]
[[2]][[1]]
[[2]][[1]][[1]]
[1] "a"

[[2]][[2]]
[[2]][[2]][[1]]
NULL

[[3]]
[[3]][[1]]
NULL

现在,我当然可以relist(unlist(b), a),但会再次丢失数据类型。什么是重组单位清单的好方法?

如果它正确处理unlist.vectors的类似属性,则会获得积分。

1 个答案:

答案 0 :(得分:1)

一种方法是:

relist2 = function(x, like, relist.vectors=F) {
    if (! relist.vectors) 
        like = rapply(a, function(f) NA, how='replace')
    lapply(relist(x, skeleton=like), function(e) unlist(e, recursive=F))
}

这保留了类并区分列表和向量:

> relist2(flatten(a), like=a)
[[1]]
[1] 1 2 3

[[2]]
[[2]][[1]]
[1] 52

[[2]][[2]]
[1] 561

[[3]]
[1] "a"

> relist2(flatten(a, unlist.vectors=T), like=a, relist.vectors=T)
[[1]]
[1] 1 2 3

[[2]]
[[2]][[1]]
[1] 52

[[2]][[2]]
[1] 561

[[3]]
[1] "a"