熊猫累积总和与分组

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