Dodawanie kolumny do ramki danych w R na podstawie pozycji innej kolumny

Question

Oto powtarzalny przykład moich danych. Na poniższej ramce danych:

df <- data.frame(Subject = c('John', 'John', 'John', 'John','Mary', 'Mary', 'Mary', 'Mary'), 
       SNR = c(-4,-4,0,4,0,4,4,8)) 

Chciałbym dodać kolumnę „rangi”, która stanowi pozycję dla SNR według tematu, tak że będzie ona wyglądać tak:

Subject SNR Rank 
John  -4 1 
John  -4 1 
John  0 2 
John  4 3 
Mary  0 1 
Mary  4 2 
Mary  4 2 
Mary  8 3 

Próbowałem przy użyciu:

dfNew <- transform(df, Rank = ave(SNR, Subject, FUN = function(x) rank(x, ties.method = "first"))) 

Ale mam następujące:

Subject SNR Rank 
John  -4 1 
John  -4 2 
John  0 3 
John  4 4 
Mary  0 1 
Mary  4 2 
Mary  4 3 
Mary  8 4 

Próbowałem również używać różnych opcji ties.method, ale żadna nie daje mi tego, czego szukam (tj. Tylko w rankingu od 1-3).

Każda pomoc będzie bardzo ceniona!

Answer 1

Korzystanie aggregate i factor w podstawowej R:

ag <- aggregate(SNR~Subject, df, function(x) as.numeric(factor(x))) 
df$rank <- c(t(ag[,-1])) 

    Subject SNR rank 
1 John -4 1 
2 John -4 1 
3 John 0 2 
4 John 4 3 
5 Mary 0 1 
6 Mary 4 2 
7 Mary 4 2 
8 Mary 8 3 

Answer 2

trochę brudna, ale wydaje się działać:

library(dplyr) 
df %>% group_by(Subject) %>% mutate(Rank = as.numeric(as.factor(SNR))) 

    Subject SNR Rank 
    <fctr> <dbl> <dbl> 
1 John -4  1 
2 John -4  1 
3 John  0  2 
4 John  4  3 
5 Mary  0  1 
6 Mary  4  2 
7 Mary  4  2 
8 Mary  8  3 

Answer 3

library(dplyr)  
df %>% 
    arrange(Subject, SNR) %>% 
    group_by(Subject) %>% 
    mutate(rank=dense_rank(SNR)) 

kredytu kurs @ bogatym-Scriven dla wspomnieć dense_rank()

Answer 4

Inna podstawowa metoda R:

transform(df1, Rank = ave(SNR, Subject, FUN = function(x) cumsum(c(TRUE, head(x, -1) != tail(x, -1))))) 

daje:

Subject SNR Rank 
1 John -4 1 
2 John -4 1 
3 John 0 2 
4 John 4 3 
5 Mary 0 1 
6 Mary 4 2 
7 Mary 4 2 
8 Mary 8 3 

Jeśli dataframe jeszcze nie zamówił, należy go zamówić najpierw df1 <- df1[order(df1$SNR),] dla tej metody, aby dać poprawny wynik.