r - 如何为感兴趣的变量中的特定值按组创建计数变量？

Question

目前，我必须在队列研究的数据收集阶段处理软件生成的paradata(长格式)。

如何通过组变量创建一个包含特定值出现次数的变量(例如 Stata 中的 by id: gen _n if VAR1==2)？

数据基本上是这样的:

 ID:   VAR1:    
 1       2      
 1       1      
 1       2      
 2       2      
 2       3      
 2       2      
 3       2      
 3       2      
 3       2       

我可以使用创建变量 count.1

`data$count.1 <- ave(data$VAR1, data$ID,  FUN = seq_along)`

 ID:   VAR1:    count.1: 
 1       2       1           
 1       1       2           
 1       2       3          
 2       2       1            
 2       3       2           
 2       2       3            
 3       2       1            
 3       2       2            
 3       2       3            

如何创建一个变量count.2，通过ID来统计VAR1中事件2发生的次数？

 ID:   VAR1:    count.1:  count.2:
 1       2       1            1
 1       1       2           NA
 1       2       3            2
 2       2       1            1
 2       3       2           NA
 2       2       3            2
 3       1       1           NA
 3       2       2            1
 3       2       3            2 

数据:

ID=c(1,1,1,2,2,2,3,3,3)
VAR1=c(2,1,2,2,3,2,1,2,2)
data <- as.data.frame(cbind(ID, VAR1))

先谢谢了!!!

Answer 1

尝试

 data$count.2 <- with(data, ave(VAR1==2, ID,
                 FUN=function(x) ifelse(x, cumsum(x), NA)) )

 data$count.2
 #[1]  1 NA  2  1 NA  2 NA  1  2

或者使用data.table

 library(data.table)
 setDT(data)[VAR1==2, count.2:=1:.N, by=ID][]
 #   ID VAR1 count.2
 #1:  1    2       1
 #2:  1    1      NA
 #3:  1    2       2
 #4:  2    2       1
 #5:  2    3      NA
 #6:  2    2       2
 #7:  3    1      NA
 #8:  3    2       1
 #9:  3    2       2

或者使用dplyr

 library(dplyr)
 data %>% 
      group_by(ID) %>% 
       mutate(count.2= ifelse(VAR1==2, cumsum(VAR1==2), NA))