purrr rbind список фреймов данных на группу - PullRequest
2 голосов
/ 27 марта 2019

После использования purrr и друзей для чтения в загрузке csvs я получил tibble, который выглядит примерно так:

library(tidyverse)

df <- 
  tibble(
    df_name = c("A", "B", "A", "A", "B"),
    data = list(iris)
  )

df

# A tibble: 5 x 2
  df_name data                  
  <chr>   <list>                
1 A       <data.frame [150 × 5]>
2 B       <data.frame [150 × 5]>
3 A       <data.frame [150 × 5]>
4 A       <data.frame [150 × 5]>
5 B       <data.frame [150 × 5]>

Я хочу rbind (или эквивалентный) все данные с общим df_name.Я хотел бы, чтобы вывод был именованным списком.Я могу сделать это с tapply:

desired = tapply(df$data, df$df_name, function(y) do.call(rbind,y))   

List of 2
 $ A:'data.frame':  450 obs. of  5 variables:
  ..$ Sepal.Length: num [1:450] 5.1 4.9 4.7 4.6 5 5.4 4.6 5 4.4 4.9 ...
  ..$ Sepal.Width : num [1:450] 3.5 3 3.2 3.1 3.6 3.9 3.4 3.4 2.9 3.1 ...
  ..$ Petal.Length: num [1:450] 1.4 1.4 1.3 1.5 1.4 1.7 1.4 1.5 1.4 1.5 ...
  ..$ Petal.Width : num [1:450] 0.2 0.2 0.2 0.2 0.2 0.4 0.3 0.2 0.2 0.1 ...
  ..$ Species     : Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
 $ B:'data.frame':  300 obs. of  5 variables:
  ..$ Sepal.Length: num [1:300] 5.1 4.9 4.7 4.6 5 5.4 4.6 5 4.4 4.9 ...
  ..$ Sepal.Width : num [1:300] 3.5 3 3.2 3.1 3.6 3.9 3.4 3.4 2.9 3.1 ...
  ..$ Petal.Length: num [1:300] 1.4 1.4 1.3 1.5 1.4 1.7 1.4 1.5 1.4 1.5 ...
  ..$ Petal.Width : num [1:300] 0.2 0.2 0.2 0.2 0.2 0.4 0.3 0.2 0.2 0.1 ...
  ..$ Species     : Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
 - attr(*, "dim")= int 2
 - attr(*, "dimnames")=List of 1
  ..$ : chr [1:2] "A" "B"

Я не могу понять, как сделать то же самое с purrr глаголами.Я думаю, что, возможно, мне нужно начать с установки имен списков:

df_p <- 
  df %>%
  mutate(data = setNames(data, df_name))

Я нашел этот вопрос, но я не могу понять, как применить в этой ситуации.

Ответы [ 3 ]

3 голосов
/ 27 марта 2019

Мы можем использовать tidyr::unnest

library(tidyverse)
df %>% split(.$df_name) %>% map(.%>%unnest() %>% select(-df_name))

#OR
df %>% split(.$df_name) %>% map(~unnest(.) %>% select(-df_name))
df %>% unnest(data) %>% split(.$df_name) 

Как @kath указал, что мы можем использовать unnest напрямую

df %>% split(.$df_name) %>% map(unnest) 
2 голосов
/ 27 марта 2019

Вы можете использовать reduce из purrr и bind_rows (аналогично rbind) из dplyr.

df_list <- df %>% 
  group_by(df_name) %>% 
  summarize(data = list(reduce(data, bind_rows)))

df_list 
# A tibble: 2 x 2
#   df_name data                  
#   <chr>   <list>                
# 1 A       <data.frame [450 x 5]>
# 2 B       <data.frame [300 x 5]>

Для той же структуры, что и в вашей tapply -версии, нам нужно добавить следующее:

df_list2 <- df_list %>% 
  split(.$df_name) %>% 
  map(~ .x$data[[1]])

str(df_list2)
List of 2
 $ A:'data.frame':  450 obs. of  5 variables:
  ..$ Sepal.Length: num [1:450] 5.1 4.9 4.7 4.6 5 5.4 4.6 5 4.4 4.9 ...
  ..$ Sepal.Width : num [1:450] 3.5 3 3.2 3.1 3.6 3.9 3.4 3.4 2.9 3.1 ...
  ..$ Petal.Length: num [1:450] 1.4 1.4 1.3 1.5 1.4 1.7 1.4 1.5 1.4 1.5 ...
  ..$ Petal.Width : num [1:450] 0.2 0.2 0.2 0.2 0.2 0.4 0.3 0.2 0.2 0.1 ...
  ..$ Species     : Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
 $ B:'data.frame':  300 obs. of  5 variables:
  ..$ Sepal.Length: num [1:300] 5.1 4.9 4.7 4.6 5 5.4 4.6 5 4.4 4.9 ...
  ..$ Sepal.Width : num [1:300] 3.5 3 3.2 3.1 3.6 3.9 3.4 3.4 2.9 3.1 ...
  ..$ Petal.Length: num [1:300] 1.4 1.4 1.3 1.5 1.4 1.7 1.4 1.5 1.4 1.5 ...
  ..$ Petal.Width : num [1:300] 0.2 0.2 0.2 0.2 0.2 0.4 0.3 0.2 0.2 0.1 ...
  ..$ Species     : Factor w/ 3 levels "setosa","versicolor",..: 1 1 1 1 1 1 1 1 1 1 ...
1 голос
/ 27 марта 2019

Я бы использовал unnest и group_split:

df %>% unnest(data) %>% group_split(df_name)

# [[1]]
# # A tibble: 450 x 6
#   df_name Sepal.Length Sepal.Width Petal.Length Petal.Width Species
#   <chr>          <dbl>       <dbl>        <dbl>       <dbl> <fct>  
# 1 A                5.1         3.5          1.4         0.2 setosa 
# 2 A                4.9         3            1.4         0.2 setosa 
# 3 A                4.7         3.2          1.3         0.2 setosa 
# 4 A                4.6         3.1          1.5         0.2 setosa 
# 5 A                5           3.6          1.4         0.2 setosa 
# 6 A                5.4         3.9          1.7         0.4 setosa 
# 7 A                4.6         3.4          1.4         0.3 setosa 
# 8 A                5           3.4          1.5         0.2 setosa 
# 9 A                4.4         2.9          1.4         0.2 setosa 
# 10 A                4.9         3.1          1.5         0.1 setosa 
# # ... with 440 more rows
# 
# [[2]]
# # A tibble: 300 x 6
#   df_name Sepal.Length Sepal.Width Petal.Length Petal.Width Species
#   <chr>          <dbl>       <dbl>        <dbl>       <dbl> <fct>  
# 1 B                5.1         3.5          1.4         0.2 setosa 
# 2 B                4.9         3            1.4         0.2 setosa 
# 3 B                4.7         3.2          1.3         0.2 setosa 
# 4 B                4.6         3.1          1.5         0.2 setosa 
# 5 B                5           3.6          1.4         0.2 setosa 
# 6 B                5.4         3.9          1.7         0.4 setosa 
# 7 B                4.6         3.4          1.4         0.3 setosa 
# 8 B                5           3.4          1.5         0.2 setosa 
# 9 B                4.4         2.9          1.4         0.2 setosa 
# 10 B                4.9         3.1          1.5         0.1 setosa 
# # ... with 290 more rows
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...