将groupby *** 作的结果插入行到原始数据框中

将groupby *** 作的结果插入行到原始数据框中

如果可以保证

X

并且

Z

仅在组中出现一次，则可以使用

groupby

and

pd.concat

*** 作：

new = df[df.col_2.isin(['X', 'Z'])]      .groupby(['col_1'], as_index=False).sum()      .assign(col_2='NEW')df = pd.concat([df, new]).sort_values('col_1')df  col_1 col_2  col_3  col_40     a     X      5      11     a     Y      3      22     a     Z      6      40     a   NEW     11      53     b     X      7      84     b     Y      4      35     b     Z      6      51     b   NEW     13     13