Добавить столбец "ранг" во фрейм данных

У меня есть датафрейм с количеством разных предметов в разные годы:

df <- data.frame(item = rep(c('a','b','c'), 3),
                 year = rep(c('2010','2011','2012'), each=3),
                 count = c(1,4,6,3,8,3,5,7,9))

И я хотел бы добавить столбец "year.rank", который дает ранг предмета в течение данного года, где более высокий счет ведет к более высокому "рангу". С учетом вышесказанного это будет выглядеть так:

  item year count year.rank
1    a 2010     1         3
2    b 2010     4         2
3    c 2010     6         1
4    a 2011     3         2
5    b 2011     8         1
6    c 2011     3         3
7    a 2012     5         3
8    b 2012     7         2
9    c 2012     9         1

Я знаю, что мог бы сделать это для всего фрейма данных, используя order(df$count), но я не уверен, как бы я сделал это к году.

5 ответов

Решение

E сть rank Функция, чтобы помочь вам с этим:

transform(df, 
          year.rank = ave(count, year, 
                          FUN = function(x) rank(-x, ties.method = "first")))
  item year count year.rank
1    a 2010     1         3
2    b 2010     4         2
3    c 2010     6         1
4    a 2011     3         2
5    b 2011     8         1
6    c 2011     3         3
7    a 2012     5         3
8    b 2012     7         2
9    c 2012     9         1

data.table версия для практики:

library(data.table)
DT <- as.data.table(df)
DT[,yrrank:=rank(-count,ties.method="first"),by=year]

   item year count yrrank
1:    a 2010     1      3
2:    b 2010     4      2
3:    c 2010     6      1
4:    a 2011     3      2
5:    b 2011     8      1
6:    c 2011     3      3
7:    a 2012     5      3
8:    b 2012     7      2
9:    c 2012     9      1

С помощью order функция,

transform(dat, x= ave(count,year,FUN=function(x) order(x,decreasing=T)))
  item year count x
1    a 2010     1 3
2    b 2010     4 2
3    c 2010     6 1
4    a 2011     3 2
5    b 2011     8 1
6    c 2011     3 3
7    a 2012     5 3
8    b 2012     7 2
9    c 2012     9 1

РЕДАКТИРОВАТЬ

Ты можешь использовать plyr здесь также:

ddply(dat,.(year),transform,x =  order(count,decreasing=T))

Используя dplyr, вы можете сделать это следующим образом:

library(dplyr) # 0.4.1
df %>% 
  group_by(year) %>% 
  mutate(yrrank = row_number(-count))

#Source: local data frame [9 x 4]
#Groups: year
#
#  item year count yrrank
#1    a 2010     1      3
#2    b 2010     4      2
#3    c 2010     6      1
#4    a 2011     3      2
#5    b 2011     8      1
#6    c 2011     3      3
#7    a 2012     5      3
#8    b 2012     7      2
#9    c 2012     9      1

Это так же, как:

df %>% 
  group_by(year) %>% 
  mutate(yrrank = rank(-count, ties.method = "first"))

Обратите внимание, что полученные данные по-прежнему сгруппированы по годам. Если вы хотите удалить группировку, вы можете просто расширить канал с помощью %>% ungroup(),

Используя ответы, данные другими, я обнаружил, что следующее работает быстрее, чем варианты transform и dyplr:

df$year.rank <- ave(count, year, FUN = function(x) rank(-x, ties.method = "first"))
Другие вопросы по тегам