python - 数据框中标志切换之间的行的总和/平均值

Question

我有这样一个数据框

---

id  power   flag
0   20      0
1   25      0
2   26      1
3   30      1
4   18      0
5   30      0
6   19      0
7   21      1
8   23      0

我正在尝试获取 power 列的行的平均值/总和输出应该是这样的

0   45 (20 +25)
1   56 (26 + 30)
2   67 (18 +30 +19)
3   21 (21)
4   23 (23)

如果标志不是二进制的怎么办？

有人可以帮我解决这些问题吗？谢谢

Answer 1

IIUC，使用 diff.ne(0) 的 bool 值的累加和作为 grouper，并且只是 sum

df.groupby(df.flag.diff().ne(0).cumsum()).power.sum()

---

flag
1    45
2    56
3    67
4    21
5    23
Name: power, dtype: int64