2011-11-21 10 views
7

obserwacji w RI mają ramkę danych z obserwacji opisanych kilka wartości, z których jeden jest czynnikiem. Posortowałem zbiór danych według tego czynnika i chciałbym dodać kolumnę, w której uzyskałabym liczbę obserwacji na każdym poziomie czynnika, np.liczba przez grupę

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

W SAS to zrobić coś podobnego:

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

Jak mogę to osiągnąć w R?

Dzięki,

Odpowiedz

13

Zastosowanie rle (run kodowania o zmiennej długości) i 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 
3

Jedno rozwiązanie wykorzystujące zasady R, zakładając, że dane są w data.frame nazwie dfr:

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

Istnieje prawdopodobieństwo lepszych rozwiązań (np zatrudniania pakiet plyr i jego ddply), ale powinno działać.

5

Oto ddply roztwór()

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)) 
}) 
Powiązane problemy