r - 在组上创建对的滚动索引

Question

我需要(使用 R)从包含组的数据集中创建对的滚动索引。考虑以下数据集:

times <- c(4,3,2)
V1 <- unlist(lapply(times, function(x) seq(1, x)))
df <- data.frame(group = rep(1:length(times), times = times), 
                 V1 = V1, 
                 rolling_index = c(1,1,2,2,3,3,4,5,5))

df
  group V1 rolling_index
1     1  1             1
2     1  2             1
3     1  3             2
4     1  4             2
5     2  1             3
6     2  2             3
7     2  3             4
8     3  1             5
9     3  2             5

我拥有的数据框包括变量组和 V1。在每个组中，V1 指定一个运行索引(可能从 1 开始，也可能不从 1 开始)。

我想创建一个看起来像 rolling_index 的新索引变量。此变量将同一组内的行和连续的 V1 值分组，从而创建一个新的滚动索引。这个新索引必须在组上是连续的。如果组内的行数不均匀(例如组 2)，则最后一个单行将获得其自己的滚动索引值。

Answer 1

你可以试试

library(data.table)
setDT(df)[,  gr:=as.numeric(gl(.N, 2, .N)), group][, 
     rollindex:=cumsum(c(TRUE,abs(diff(gr))>0))][,gr:= NULL]
#    group V1 rolling_index rollindex
#1:     1  1             1         1
#2:     1  2             1         1
#3:     1  3             2         2
#4:     1  4             2         2
#5:     2  1             3         3
#6:     2  2             3         3
#7:     2  3             4         4
#8:     3  1             5         5
#9:     3  2             5         5

或者使用 base R

 indx1 <- !duplicated(df$group)
 indx2 <- with(df, ave(group, group, FUN=function(x)
                           gl(length(x), 2, length(x))))
 cumsum(c(TRUE,diff(indx2)>0)|indx1)
 #[1] 1 1 2 2 3 3 4 5 5

更新

以上方法均基于“组”列。假设您已经有一个按组的序列列('V1')，如示例所示，创建滚动索引更容易

 cumsum(!!df$V1 %%2)
 #[1] 1 1 2 2 3 3 4 5 5

如帖子中所述，如果某些组的'V1'列不是从'1'开始，我们可以从'group'中获取序列，然后像上面一样执行cumsum

 cumsum(!!with(df, ave(seq_along(group), group, FUN=seq_along))%%2)
 #[1] 1 1 2 2 3 3 4 5 5