Pandas累加和多个分组依据

Question

我有一个表，其中有每天，每个州，每个县的计数

df = pd.DataFrame.from_dict({'Date': [1,1,1,1,2,2,2,2,2,3], 'state': ['CA','CA','IN','IN','CA','CA','IN','IN','TX','CA'], 'County':['San Diego','SFO','Monroe','Kent','San Diego','SFO','Monroe','Hous','Mazs','San Diego'], 'count':[1,1,1,1,2,2,2,2,3,4]})

​

​

我想把它转换成

​

我像这样试过了

data.groupby(['DATE','State','County'])['count'].transform(pd.Series.cumsum)

我没有得到累加和……我得到了初始表作为结果

Answer 1

你可以做到

data['new'] = data.groupby(['State','County'])['count'].cumsum()