esiste un equivalente della funzione egen di Stata?

Question

Stata ha un comando molto bello, egen, che semplifica il calcolo delle statistiche rispetto al gruppo di osservazione. Ad esempio, è possibile calcolare il massimo, la media e il minimo per ciascun gruppo e aggiungerli come variabile nel set di dati dettagliato. Il comando di Stata è una riga di codice:

by group : egen max = max(x) 

non ho mai trovato lo stesso comando in R. summarise nel pacchetto dplyr lo rende facile da calcolare le statistiche per ogni gruppo, ma poi devo correre un ciclo per associare la statistica ad ogni osservazione:

library("dplyr") 
N <- 1000 
tf <- data.frame(group = sample(1:100, size = N, replace = TRUE), x = rnorm(N)) 
table(tf$group) 
mtf <- summarise(group_by(tbl_df(tf), group), max = max(x)) 
tf$max <- NA 
for (i in 1:nrow(mtf)) { 
    tf$max[tf$group == mtf$group[i]] <- mtf$max[i] 
} 

Qualcuno ha una soluzione migliore?

Answer 1

Ecco alcuni approcci:

dplyr

library(dplyr) 

tf %>% group_by(group) %>% mutate(max = max(x)) 

ave

Questo utilizza solo la base di R:

transform(tf, max = ave(x, group, FUN = max)) 

data.table

library(data.table) 

dt <- data.table(tf) 
dt[, max:=max(x), by=group]