python - 在 Pandas 数据框中成对计数

Question

我正在 Pandas 中寻找一种简单的方法来计算两列之间的所有成对组合。例如:

df

Col1    Col2   Freq
 A        B     1
 B        A     1
 A        C     1
 C        B     1
 C        A     1

预期输出:

Col1    Col2   Freq
 A        B     2
 A        C     2
 C        B     1

谢谢

Answer 1

你可以试试:

df2=(df
  .groupby(
       df[["Col1", "Col2"]]
       .apply(lambda x: str(sorted(x)), axis=1)
       )
  .agg({"Col1": "first", "Col2": "first", "Freq": "sum"})
  .reset_index(drop=True)
)

输出:

  Col1 Col2  Freq
0    A    B     2
1    A    C     2
2    C    B     1