tidyr相当于reshape2 :: melt的向量的深层嵌套列表

编程问答 2022-08-13

问题描述

使用reshape2，很容易将向量的深层嵌套列表转换为长的data.frame，其中包含有关每个值的原始列表位置的信息。

# generate nested list of depth 3 with 2 branches at each level
n_l <- purrr::array_tree(
  array(1:2^3,rep_len(2,3))
)

m_n_l <- reshape2::melt(n_l)

# this provides a df of values where original list position is labelled using 
# [n_dimensions] columns each with [n_branches] values. yay! that's what I want
m_n_l

#    value L3 L2 L1
# 1      1  1  1  1
# 2      5  2  1  1
# 3      3  1  2  1
# 4      7  2  2  1
# 5      2  1  1  2
#        ...

# [reshape2::melt() also handles the case where leaf node vectors have 
# arbitrary number of elements]
reshape2::melt(rapply(n_l,function(x) x * 1:sample(1:3,1)))

reshape2现在已经退休，并且鼓励其用户使用tidyr。但是，我找不到一种tidyr的方法来复制reshape2::melt()的上述功能。 pivot_longer()，hoist()和unnest()似乎是取代melt()的函数，但是它们似乎专用于data.frame或data.frame列表的特定情况。

tidyr可以这样做吗？

解决方法

使用rrapply的一个选项可能是：

Reduce(rbind,rrapply(n_l,f = function(x,.xpos) c(.xpos,x),how = "flatten"))

     [,1] [,2] [,3] [,4]
init    1    1    1    1
        1    1    2    5
        1    2    1    3
        1    2    2    7
        2    1    1    2
        2    1    2    6
        2    2    1    4
        2    2    2    8

但是，如果您正在寻找特定的tidyverse选项，则可能不是一个很紧凑的选项：

enframe(n_l) %>%
 mutate(value = map(value,~ enframe(.,name = "name2"))) %>%
 unnest(value) %>%
 mutate(value = map(value,name = "name3"))) %>%
 unnest(value) %>%
 mutate(value = unlist(value))

   name name2 name3 value
  <int> <int> <int> <int>
1     1     1     1     1
2     1     1     2     5
3     1     2     1     3
4     1     2     2     7
5     2     1     1     2
6     2     1     2     6
7     2     2     1     4
8     2     2     2     8
rrreshape2reshape2tidyr