Номер наблюдения по группам

Question

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

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

В SAS я это делаю что-то вроде:

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

Как я могу добиться того, что в R?

Спасибо,

Answer 1

Использование rle (Run Length кодирования) и 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

Одно из решений с использованием базы R, предполагая, что ваши данные в data.frame имени dfr:

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

Есть, вероятно, более эффективные решения (например, с использованием пакета plyr и ее ddply), но должно сработать.

Answer 3

Здесь ddply раствор()

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)) 
})