Create Nested List from Tibble

Question

I'd like to create a nested listed from a tibble, so the data is stored more efficiently and it is easier to extract information with subsetting.

I have a 3 column tibble like the following:

library(dplyr)

df <- tribble(
~a,~b,~c,
"a1", "b1", 1,
"a1", "b1", 2,
"a1", "b2", 1, 
"a1", "b2", 2,
)

I'd like to convert this to a list identical to this:

list(a1 = list(b1 = 1:2, b2 = 1:2))

Does anyone know how I can do this please?

Thanks

score 3 · Answer 1 · answered Dec 06 '21 at 13:54

3

Split twice, i.e.

lapply(split(df,df$a), function(x) split(x$c,x$b))

$a1
$a1$b1
[1] 1 2

$a1$b2
[1] 1 2

answered Dec 06 '21 at 13:54

Sotos

51,121
6
32
66

Thanks! That's great. Is there a way to make this generalizable (e.g. if you had 4 or more columns, ordered left to right in how you wanted the nesting to be made)? – Liam Wright Dec 06 '21 at 14:10
Not that easily I reckon – Sotos Dec 06 '21 at 14:11

PaulS · Answer 2 · 2021-12-06T16:14:18.283

2

Something like the following?

library(tidyverse)

df <- tribble(
  ~a,~b,~c,
  "a1", "b1", 1,
  "a1", "b1", 2,
  "a1", "b2", 1, 
  "a1", "b2", 2,
)

df %>%
  group_by(a,b) %>% group_split() %>% map(~ list(.x$c) %>% set_names(.x$b[1])) %>% 
  unlist(recursive = F) %>% list %>% set_names(unique(df$a))

#> $a1
#> $a1$b1
#> [1] 1 2
#> 
#> $a1$b2
#> [1] 1 2

edited Dec 06 '21 at 16:14

answered Dec 06 '21 at 13:51

PaulS

21,159
2
9
26

Thanks. Thought of that, but I wanted a list of vectors are the end, rather than another tibble. – Liam Wright Dec 06 '21 at 14:10
I have edited my solution, in order to get only a list, as you want, @LiamWright. – PaulS Dec 06 '21 at 16:16

score 1 · Answer 3 · answered Dec 07 '21 at 08:31

rrapply() in the package rrapply has an option how = "unmelt" to convert melted data.frames into nested lists (which also works for arbitrary levels of nesting).

Each row in the input data.frame is converted to a single node path in the output nested list:

library(dplyr)
library(rrapply)

## 1 row -> 1 leaf
rrapply(df, how = "unmelt")
#> $a1
#> $a1$b1
#> [1] 1
#> 
#> $a1$b1
#> [1] 2
#> 
#> $a1$b2
#> [1] 1
#> 
#> $a1$b2
#> [1] 2

To assign multiple rows to a single leaf, we can nest the c column first:

## 2 rows -> 1 leaf
df %>% 
  group_by(a, b) %>%
  summarize(c = list(c), .groups = "drop") %>%
  rrapply(how = "unmelt")
#> $a1
#> $a1$b1
#> [1] 1 2
#> 
#> $a1$b2
#> [1] 1 2

Create Nested List from Tibble

3 Answers3