r - 向数据框添加一个场合标志

Question

对于每个人，当金额大于零时，我想为我的数据框添加一个场合标志。我需要这个标志进行进一步的计算。这是我想要实现的目标。

dfin <- 

ID AMT 
1  50
1  NA
1  10
1  NA
2  15
2  NA
2  NA
3  10
3  15

dfout <- 

ID  AMT FLAG
1   50   1
1   NA   1
1   10   2
1   NA   2
2   15   1
2   NA   1
2   NA   1
3   10   1
3   15   2

我如何在 R 中实现这一点？

Answer 1

您可以测试哪些值不是 NA 并计算累积和。

dfout = dfin
dfout$FLAG = cumsum(!is.na(dfin$AMT))
dfout
  ID AMT FLAG
1  1  50    1
2  1  NA    1
3  1  10    2
4  1  NA    2
5  2  15    3
6  2  NA    3
7  2  NA    3
8  3  10    4