r - 列中的每次更改增加 1

Question

假设我有以下数据框

set.seed(123)
df <- data.frame(var1=(runif(10)>0.5)*1)

var1可以有任何类型/数量的级别，而不是 0 和 1s

我想创建一个 var2每次增加 1 var1变化 没有 使用 for loop
这种情况下的预期结果是:

data.frame(var1=(runif(10)>0.5)*1, var2=c(1, 2, 3, 4, 4, 5, 6, 6, 6, 7))

var1 var2
   0    1
   1    2
   0    3
   1    4
   1    4
   0    5
   1    6
   1    6
   1    6
   0    7

数据框的另一个选项可能是:

df <- data.frame(var1=c("a", "a", "1", "0", "b", "b", "b", "c", "1", "1"))

在这种情况下，结果应该是:

var1 var2
   a    1
   a    1
   1    2
   0    3
   b    4
   b    4
   b    4
   c    5
   1    6
   1    6

Answer 1

基于弗利克先生的回答:

df$var2 <- cumsum(c(0,as.numeric(diff(df$var1))!=0))

但如果你不想使用 diff你仍然可以使用:

df$var2 <- c(0,cumsum(as.numeric(with(df,var1[1:(length(var1)-1)] != var1[2:length(var1)]))))

它从 0 开始，而不是从 1 开始，但我相信如果您愿意，您会看到如何更改它。