python - 当使用 pandas 连续出现多个组时，如何按组添加数字

Question

我有一个像这样的数据框:

df
col1    col2
 12       A
 14       A
 22       B
 24       C
 20       A
 18       B
 16       B

现在，如果 col2 值连续出现一次以上，我想在 col2 的基础上添加 col1 的值。最终的数据框应如下所示:

col1    col2
 A       26
 B       22
 C       24
 A       20
 B       34

我可以使用groupby()，但如何区分连续条件？

Answer 1

使用GroupBy.agg与 Series.ne 创建的助手 Series与 Series.shift和 Series.cumsum :

s = df['col2'].ne(df['col2'].shift()).cumsum()
df = df.groupby(s).agg({'col2':'first', 'col1':'sum'}).reset_index(drop=True)
print (df)
  col2  col1
0    A    26
1    B    22
2    C    24
3    A    20
4    B    34