R dplyr повторить строки данных кадра по группе

Question

У меня есть датафрейм, содержащий группы. Один столбец ntimes указывает, сколько раз группа должна повторяться, сохраняя один и тот же порядок строк. Я хочу расширить свой массив данных группами строк, повторяющимися несколько раз. Однако я не уверен, как сделать это эффективно?

Мой пример:

  # Repeat rows by group
set.seed(5)
df <- data.frame(my.group = rep(c("a", "b", "z"), each = 3),
                 vals = runif(9),
                 ntimes = c(3,3,3, 1,1,1,2,2,2))

  my.group      vals ntimes
1        a 0.1104530      3
2        a 0.2732849      3
3        a 0.4905132      3
4        b 0.3184040      1
5        b 0.5591728      1
6        b 0.2625931      1
7        z 0.2018752      2
8        z 0.3875257      2
9        z 0.8878698      2

Группу a следует повторить 3 раза, группу b один раз, группу z 3 раза.

Ожидаемый результат:

  my.group      vals ntimes
1        a 0.1104530      3
2        a 0.2732849      3
3        a 0.4905132      3
4        a 0.1104530      3
5        a 0.2732849      3
6        a 0.4905132      3
7        a 0.1104530      3
8        a 0.2732849      3
9        a 0.4905132      3
10       b 0.3184040      1
11       b 0.5591728      1
12       b 0.2625931      1
13       z 0.2018752      2
14       z 0.3875257      2
15       z 0.8878698      2
16       z 0.2018752      2
17       z 0.3875257      2
18       z 0.8878698      2

Я пробовал несколько подходов с использованием dplyr, но ни один из них не дает желаемых результатов:

# repeat df rows by group
library(dplyr)
df %>% 
  group_by(my.group) %>% 
  mutate(new = rep(seq_len(n()/2), each = 2, length.out = n()))


df %>% 
  group_by(my.group) %>% 
  slice(rep(1:n(), each = 2))


df %>% 
  group_by(my.group) %>% 
  mutate(count = c(3,1,2)) %>% 
  expand(ntimes = seq(1:ntimes))

Я высоко ценю любые предложения.

Answer 1

Дайте только одно значение аргументу times в rep. Поскольку вы хотите сделать это по группам, вы можете использовать любое значение из столбца ntimes.

library(dplyr)
df %>% group_by(my.group) %>% slice(rep(1:n(), first(ntimes)))
#Similar other variations could be
#df %>% group_by(my.group) %>% slice(rep(seq_len(n()), first(ntimes)))
#df %>% group_by(my.group) %>% slice(rep(seq_along(ntimes), first(ntimes)))

#  my.group  vals ntimes
#  <fct>    <dbl>  <int>
# 1 a        0.110      3
# 2 a        0.273      3
# 3 a        0.491      3
# 4 a        0.110      3
# 5 a        0.273      3
# 6 a        0.491      3
# 7 a        0.110      3
# 8 a        0.273      3
# 9 a        0.491      3
#10 b        0.318      1
#11 b        0.559      1
#12 b        0.263      1
#13 z        0.202      2
#14 z        0.388      2
#15 z        0.888      2
#16 z        0.202      2
#17 z        0.388      2
#18 z        0.888      2

---

Делать это в базе R удивительно сложно или, может быть, есть способ, который я не могу понять

df[unlist(Map(rep, split(1:nrow(df), df$my.group), 
                   tapply(df$ntimes, df$my.group, head, 1))), ]

данные

df <- structure(list(my.group = structure(c(1L, 1L, 1L, 2L, 2L, 2L, 
3L, 3L, 3L), .Label = c("a", "b", "z"), class = "factor"), vals = c(0.110453, 
0.2732849, 0.4905132, 0.318404, 0.5591728, 0.2625931, 0.2018752, 
0.3875257, 0.8878698), ntimes = c(3L, 3L, 3L, 1L, 1L, 1L, 2L, 
2L, 2L)), class = "data.frame", row.names = c("1", "2", "3", 
"4", "5", "6", "7", "8", "9"))