python - Pandas 中的列组添加

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 值的组，计算总和，然后重新组合数据帧？

标签： pythonpandas

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