python - Pandas 根据条件合并两行

Question

关闭。此题需要 details or clarity 。目前不接受答案。

---

想要改进这个问题吗？通过 editing this post 添加详细信息并澄清问题.

已关闭 2 年前。

Improve this question

我想知道是否有一种方法可以按行分组然后条件聚合组。谢谢
例如，因为 colA 是 A，应该将第二个 A 的 colB 设置为 test1 并求和 colC

colA colB colC
A   test1  5
A   None   6
B   test1  4
C   test1  5
C   test3  4
D   test4  5
D   None   4

预期结果

colA colB colC
A   test1  11
B   test1  4
C   test1  5
C   test3  4
D   test4  9

Answer 1

IIUC，您可以使用 groupby 和 fill，然后再次使用 group 和 sum

df['colB'] = df.groupby('colA').colB.ffill().bfill()
df.groupby(['colA','colB'], as_index = False).sum()

    colA    colB    colC
0   A       test1   11
1   B       test1   4
2   C       test1   5
3   C       test3   4
4   D       test4   9