最近在做数据处理的时候遇到了一些问题,需要将一个dataframe按照一定的条件groupby之后,统计每个group的长度,并在原始的dataframe中创建一个新的列。
原始dataframe:
我想要的dataframe,按照is_decrease和crossing分组之后,原始的dataframe新增一列,显示每个分组的长度:
df['count'] = df.groupby(['is_decrease', 'crossing'])['id'].transform('count')