在同一个数据帧中使用Pandas进行求和和累计求和

Question

我使用以下代码创建了一个总和和累加值，但它们分别在两个不同的数据框中。我希望它们能够合并到一个数据框中。

asp = np.array(np.array([0,0,1]))
asq = np.array(np.array([10,10,20]))
columns=['asp']
df = pd.DataFrame(asp, index=None, columns=columns)
df['asq'] = asq
df = df.groupby(by=['asp']).sum()
dfcum =df.cumsum()

如何在同一个数据框中同时计算总和和累加和。不清楚如何实现这一点。以下是我想要的结果。

     asqsum cumsum
asp     
0     20      20
1     20      40

- Tampa

Answer 1

也许你想要这个？

In [20]: df['asq_cum']=df['asq'].cumsum()

In [21]: df
Out[21]: 
     asq  asq_cum
asp              
0     20       20
1     20       40

- Nipun Batra

---

您是真正的MVP。 - Sam