Número de observación por grupo

Question

En R Tengo un marco de datos con observaciones descritas por varios valores, uno de los cuales es un factor. He ordenado el conjunto de datos por este factor y me gustaría agregar una columna en la que obtendría una cantidad de observaciones en cada nivel del factor, p.

factor obsnum 
a  1 
a  2 
a  3 
b  1 
b  2 
b  3 
b  4 
c  1 
c  2 
... 

En SAS lo hago con algo como:

data logs.full; 
    set logs.full; 
    count + 1; 
    by cookie; 
    if first.cookie then count = 1; 
run; 

¿Cómo puedo lograr que en R?

Gracias,

Answer 1

Uso rle (ejecutar la codificación de longitud) y sequence:

x <- c("a", "a", "a", "b", "b", "b", "b", "c", "c") 

data.frame(
    x=x, 
    obsnum = sequence(rle(x)$lengths) 
) 

    x obsnum 
1 a  1 
2 a  2 
3 a  3 
4 b  1 
5 b  2 
6 b  3 
7 b  4 
8 c  1 
9 c  2 

Answer 2

Una solución utilizando base de R, suponiendo que los datos están en un data.frame llamado dfr:

dfr$cnt<-do.call(c, lapply(unique(dfr$factor), function(curf){ 
    seq(sum(dfr$factor==curf)) 
})) 

es probable mejores soluciones (por ejemplo, el empleo de paquete plyr y su ddply), pero Deberia trabajar.

Answer 3

Aquí es el ddply solución()

dataset <- data.frame(x = c("a", "a", "a", "b", "b", "b", "b", "c", "c")) 
library(plyr) 
ddply(dataset, .(x), function(z){ 
    data.frame(obsnum = seq_along(z$x)) 
})