同一数据框中的Pandas、sum和cum sum

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()

如何将sum和cumsum放在同一数据帧中。完全不知道怎么做。下面是我想要的

     asqsum cumsum
asp     
0     20      20
1     20      40

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