按组划分的观察数量

Question

在 RI 中，有一个数据框，其中的观察结果由多个值描述，其中一个值是一个因子。我已经按这个因素对数据集进行了排序，并且想添加一列，在其中我可以在该因素的每个级别上获得一些观察结果，例如

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

在 SAS 中，我可以这样做：

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

如何在 R 中实现这一点？

谢谢，

Answer 1

使用rle（行程编码）和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

Answer 2

这是 ddply() 解决方案

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))
})

Answer 3

一种使用基础 R 的解决方案，假设您的数据位于名为 dfr 的 data.frame 中：

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

可能有更好的解决方案（例如，使用包 plyr 和它的 ddply ），但它应该可以工作。