如何在条件中添加pandas中的行

时间:2014-11-11 13:19:27

标签: python pandas dataframe

也许有人知道如何通过分组特定条件来添加数据帧的两行。

dfa.groupby(['Col1','Col2'])[['Quantity']].sum()

说我们有这个df:

  Col1  Col2 Quantity
0  1     1     10 
1  1     1     10
2  2     1     3
3  1     2     3
4  1     2     3

我试图得到这个: 总和条件: 一行的Col1元素等于Col1 AND中其他行的元素 该行的Col2元素等于Col2中另一行的元素

  Col1  Col2 Quantity
0  1     1     20 
2  2     1     3
3  1     2     6

2 个答案:

答案 0 :(得分:1)

这看起来像你在寻找:

dfa[dfa.Col1 == dfa.Col2].groupby(['Col1','Col2'])[['Quantity']].sum()

答案 1 :(得分:0)

我认为Groupby可以做到这一点。