如何根据另一列定义的组计算一列的排名?

时间:2011-04-13 11:54:38

标签: r algorithm statistics dataset

R版本2.11.1 Windows 7上的32位

我得到如下数据集:

USER_A USER_B SCORE
1        6      0.2
1        7      0.1
1        10     0.15
2        6      0.2
2        9      0.12
3        8      0.15
3        9      0.3

USER_A为1:3,USER_B为6:10。现在我需要通过SCORE输出USER_A的USER_B排名:

USER_A      ranking of USER_B
1  3  1  2  #the ranking of USER_B 6,7,10(which belong to USER_A 1)
2  2  1     #the ranking of USER_B 6,9(which belong to USER_A 2)
3  1  2     #the ranking of USER_B 8,9(which belong to USER_A 3)
事实上,我只需要输出排名:

3 1 2
2 1
1 2

令人不安,因为每行的长度不同!我无法将它们存储在矩阵中然后输出它们。

有人能帮我解决这个问题吗?

1 个答案:

答案 0 :(得分:2)

df <- read.table(con <- textConnection("USER_A USER_B SCORE
1        6      0.2
1        7      0.1
1        10     0.15
2        6      0.2
2        9      0.12
3        8      0.15
3        9      0.3
"), header = TRUE)
close(con)

一种方法是拆分数据:

sdf <- with(df, split(SCORE, f = USER_A))
lapply(sdf, rank)

最后一行给出:

> lapply(sdf, rank)
$`1`
[1] 3 1 2

$`2`
[1] 2 1

$`3`
[1] 1 2

另一种方法是使用aggregate(),如下所示:

aggregate(SCORE ~ USER_A, data = df, rank)

返回:

> (foo <- aggregate(SCORE ~ USER_A, data = df, rank))
  USER_A   SCORE
1      1 3, 1, 2
2      2    2, 1
3      3    1, 2

但是这里的输出有点不同,现在我们有一个数据框,第二个组件SCORE是一个列表,就像输出的lapply()版本一样:

> str(foo)
'data.frame':   3 obs. of  2 variables:
 $ USER_A: int  1 2 3
 $ SCORE :List of 3
  ..$ 0: num  3 1 2
  ..$ 1: num  2 1
  ..$ 2: num  1 2
> foo$SCORE
$`0`
[1] 3 1 2

$`1`
[1] 2 1

$`2`
[1] 1 2