Репликация каждую строку data.frame и указать количество повторений для каждой строки

Question

df <- data.frame(var1=c('a', 'b', 'c'), var2=c('d', 'e', 'f'), freq=1:3) 

Что самый простой способ расширить первые две колонны выше data.frame, так что каждая строка отображается количество раз указанном в колонке «freq»?

Другими словами, перейти от этого:

df 
    var1 var2 freq 
1 a d 1 
2 b e 2 
3 c f 3 

к этому:

df.expanded 
    var1 var2 
1 a d 
2 b e 
3 b e 
4 c f 
5 c f 
6 c f 

Answer 1

Вот одно решение:

df.expanded <- df[rep(row.names(df), df$freq), 1:2] 

Результат:

var1 var2 
1  a d 
2  b e 
2.1 b e 
3  c f 
3.1 c f 
3.2 c f 

Answer 2

@ Решение neilfws отлично подходит для data.frame s, но не для data.table s, так как им не хватает row.names. Этот подход работает для обоих:

df.expanded <- df[rep(seq(nrow(df)), df$freq), 1:2] 

Для data.table хотя вам нужно добавить with=F и опционально может удалить df$:

dt <- data.table(df) 
dt.expanded <- dt[rep(seq(.N), freq), !"freq", with=F] 

Answer 3

Использование expandRows() из splitstackshape пакета:

library(splitstackshape) 
expandRows(df, "freq") 

Простой синтаксис, очень быстрый, работает на data.frame или data.table.

Результат:

var1 var2 
1  a d 
2  b e 
2.1 b e 
3  c f 
3.1 c f 
3.2 c f 

Answer 4

В случае, если вам нужно сделать эту операцию на очень больших data.frames я рекомендовал бы преобразовать его в data.table и использовать следующие, которые должны работать намного быстрее:

library(data.table) 
dt <- data.table(df) 
dt.expanded <- dt[ ,list(freq=rep(1,freq)),by=c("var1","var2")] 
dt.expanded[ ,freq := NULL] 
dt.expanded 

Смотрите, насколько быстрее это решение:

df <- data.frame(var1=1:2e3, var2=1:2e3, freq=1:2e3) 
system.time(df.exp <- df[rep(row.names(df), df$freq), 1:2]) 
## user system elapsed 
## 4.57 0.00 4.56 
dt <- data.table(df) 
system.time(dt.expanded <- dt[ ,list(freq=rep(1,freq)),by=c("var1","var2")]) 
## user system elapsed 
## 0.05 0.01 0.06 

Answer 5

старый вопрос, новый глагол в tidyverse:

library(tidyr) # version >= 0.8.0 
df <- data.frame(var1=c('a', 'b', 'c'), var2=c('d', 'e', 'f'), freq=1:3) 
df %>% 
    uncount(freq) 

е