我希望获得一个类别内的序列。
我的数据如下:
A B
1 1
1 2
1 2
1 3
1 3
1 3
1 4
1 4
我希望能够获取变量"c",使我的数据看起来像这样:
A B C
1 1 1
1 2 1
1 2 2
1 3 1
1 3 2
1 3 3
1 4 1
1 4 2
使用seq_along
和ave
:
> mydf$C <- with(mydf, ave(A, A, B, FUN = seq_along))
> mydf
A B C
1 1 1 1
2 1 2 1
3 1 2 2
4 1 3 1
5 1 3 2
6 1 3 3
7 1 4 1
8 1 4 2
如果你的数据已经按顺序排列(在这种情况下是这样的),你也可以使用 rle
和 sequence
(mydf$C <- sequence(rle(do.call(paste, mydf))$lengths)
),但是用 ave
没有这个限制。如果你喜欢使用 data.table
,你可以按如下方式利用 .N
:
library(data.table)
DT <- data.table(mydf)
DT[, C := sequence(.N), by = c("A", "B")]
DT
# A B C
# 1: 1 1 1
# 2: 1 2 1
# 3: 1 2 2
# 4: 1 3 1
# 5: 1 3 2
# 6: 1 3 3
# 7: 1 4 1
# 8: 1 4 2