熊猫中的列组添加

Question

我正在使用这样的数据框：

state current
  0     0
  0     0
  0     0
  0     0
  1     13
  1     7
  1     1
  1     12
  0     0
  0     0
  0     0
  1     8
  1     9

我想找到状态（和电流）不为0的电流组的总和，并为该组中的所有条目写入该总和。理想情况下，我会得到这样的结果：

state current sum
  0     0      0
  0     0      0
  0     0      0
  0     0      0
  1     13     33
  1     7      33
  1     1      33
  1     12     33
  0     0      0
  0     0      0
  0     0      0
  1     8      17
  1     9      17

这样做是最好的方法，将数据帧分成值不为0的组，计算总和，然后重新组装数据帧吗？

Answer 1

当您groupby时，首先检查groupby键是否为0，然后我们执行cumsum

df.groupby(df.state.eq(0).cumsum()).current.transform('sum')
0      0
1      0
2      0
3     33
4     33
5     33
6     33
7     33
8      0
9      0
10    17
11    17
12    17
Name: current, dtype: int64