Добавить столбец "ранг" во фрейм данных
У меня есть датафрейм с количеством разных предметов в разные годы:
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"))