Python 熊猫及时总结

我有如下数据：

ID  A   Date        B
1   5   2019-01-01  4
1   5   2019-01-03  6
1   5   2019-01-15  8
2   9   2019-01-02  2
2   9   2019-01-18  3

我不会是：

[结果C栏]

ID  A   Date        B   **C**
1   5   2019-01-01  4   **9**
1   5   2019-01-03  6   **15**
1   5   2019-01-15  8   **23**
2   9   2019-01-02  2   **11**
2   9   2019-01-18  3   **14**

我用for循环解决了这个问题，但按时间执行，2M行是不可接受的

有谁有更好的方法吗？感谢所有想要帮助的人：）

使用：

---

这是一个伟大的梦想！谢谢！

C[0] = A[0] + B[0] (9)

C[1] = A[1] + B[0] + B[1] (where B[0] + B[1] is C[1]=15)

C[0] = A[2] + B[0] + B[1] + B[2] (where B[0] + B[1] + B[2] is C[2]=23)

C[0] = A[3] + B[3] 

C[0] = A[4] + B[3] + B[4]

In [544]: df['C'] = df.groupby('ID').B.cumsum() + df.A

In [545]: df
Out[545]: 
   ID  A        Date  B   C
0   1  5  2019-01-01  4   9
1   1  5  2019-01-03  6  15
2   1  5  2019-01-15  8  23
3   2  9  2019-01-02  2  11
4   2  9  2019-01-18  3  14