创建一个计数变量，我可以用来将我的失业数据分组到R中

Question

我有以下数据，我创建了变量＆＃34; B＆＃34;功能：

index <- which(Count$unemploymentduration ==1)
Count$B[index]<-1:length(index)

ID unemployment B
1    1          1 
1    2          NA
1    3          NA 
1    4          NA 
2    1          2 
2    2          NA 
2    0          NA
2    1          3
2    2          NA
2    3          NA
2    4          NA
2    5          NA

我想以这种方式获取数据并且不知道如何得到这样的数据。 思考一个＆＃34; if-function＆＃34;但从未在R中使用过。

ID unemployment B
1    1          1 
1    2          1
1    3          1
1    4          1
2    1          2 
2    2          2 
2    0          2
2    1          3
2    2          3
2    3          3
2    4          3
2    5          3

有人可以帮帮我吗？

Answer 1

我们可以使用na.locf

中的library(zoo)

library(zoo)
Count$B <- na.locf(Count$B)

但是，这可以直接创建，而无需使用＆＃39;索引＆＃39;

Count$B <- cumsum(Count$unemployment==1)