STATA分段求和

数据示例如下：A列有0、1、2三种，我想求A列的每段1和2之间的B列的sum，这个结果填到C列里。
  A  B  C
   0  0  0
   1  2  2
   0  4  6
   0  5  11
   0  3  14
   2  0  0
   0  0  0
   0  0  0
   1  3  3
   0  6  9
   0  7  16
   2  0  0

现在的C列是我手工计算的结果，就是想要得到的结果。

请大神不吝赐教，十分感谢！

1. gen D = sum(A)
   
2. bys D: gen C = sum(B)
   
3. drop D

复制代码

gen order = _n
gen id2 = sum(A==1) if sum(A==1) != sum(A==2)
bys id2 (order) : gen wanted = sum(B) if !missing(id2)
sort order
list, sepby(id2)