python - 如何将 Pandas 中不同列中具有相同项目的行添加在一起

Question

我有一个示例数据框，如下所示。我想最终将第 1 行和第 3 行组合在一起，因为它们在不同的列中包含相同的项目。

x    y      count
a,b  b,a      5
a,c  c,a      2
b,a  a,b      1

我花了很多时间试图解决这个问题，但还没有遇到好的解决方案。我应该采取什么步骤来达到下面的最终数据框？

x    y      count
a,b  b,a      5+1
a,c  c,a      2

Answer 1

你可以试试:

df.groupby((df.x + df.y).str.replace(',', '').apply(lambda x: ''.join(sorted(x)))
           ).agg({'x': 'first', 'y': 'first', 'count': sum}).reset_index(drop=True)

输出:

     x    y  count
0  a,b  b,a      6
1  a,c  c,a      2