Выборка разного количества строк по группам в dplyr tidyverse

Я хотел бы выбрать строки из фрейма данных по группам. Но вот в чем загвоздка: я хотел бы выбрать другое количество записей на основе данных из другой таблицы. Вот мои воспроизводимые данные:

df <- data_frame(
  Stratum = rep(c("High","Medium","Low"), 10),
  id = c(1:30),
  Value = runif(30)
)

sampleGuide <- data_frame(
  Stratum = c("High","Medium","Low"),
  Surveys = c(3,2,5)
)

Вывод должен выглядеть так:

# A tibble: 10 × 2
   Stratum      Value
     <chr>      <dbl>
1     High 0.21504972
2     High 0.71069005
3     High 0.09286843
4   Medium 0.52553056
5   Medium 0.06682459
6      Low 0.38793128
7      Low 0.01285081
8      Low 0.87865734
9      Low 0.09100829
10     Low 0.14851919

Вот моя НЕРАБОЧАЯ попытка

> df %>% 
+   left_join(sampleGuide, by = "Stratum") %>% 
+   group_by(Stratum) %>% 
+   sample_n(unique(Surveys))
Error in unique(Surveys) : object 'Surveys' not found

Также

> df %>% 
+   group_by(Stratum) %>% 
+   nest() %>% 
+   left_join(sampleGuide, by = "Stratum") %>% 
+   mutate(sample = map(., ~ sample_n(data, Surveys)))
Error in mutate_impl(.data, dots) : 
      Don't know how to sample from objects of class function

Похоже, что sample_n требует, чтобы size было одним числом. Любые идеи?

Я ищу только tidyverse решений. Дополнительные баллы за purrr!

Это была похожая проблема, но я не удовлетворен принятым ответом, потому что в IRL количество слоев, с которыми я имею дело, велико.


r random dplyr purrr tidyr
person Zafar    schedule 15.01.2017    source источник

Ответы (1)


arrow_upward
8
arrow_downward

Разобрался с map2() из purrr

df %>% 
  nest(-Stratum) %>% 
  left_join(sampleGuide, by = "Stratum") %>% 
  mutate(Sample = map2(data, Surveys, sample_n)) %>% 
  unnest(Sample)
person Zafar    schedule 16.01.2017