熊猫和sum和cum在同一数据帧中

时间:2014-01-01 11:39:51

标签: python pandas

我使用下面的内容来创建一个总和和一个cumsum。但它们分为两个独立的数据帧。我想要一个人

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

1 个答案:

答案 0 :(得分:14)

也许你想要这个?

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

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