Pandas：如何根据不同列的值对列元素的组合进行分组以指示共现？

我有一个以下形式的 pandas 数据框df,

Batch_ID    Product_ID
   1            A
   1            B
   1            C
   2            B
   2            B
   2            C
   2            C
   3            B
   3            B
   3            C
   4            C
   4            D
   5            D

我想从中获得一个边缘列表，本质上是一个新的数据框边列表df（我无法将其转换为 networkx 对象）的形式，

Source       Target         Weight
  A             B             1.0
  A             C             1.0
  A             D             0.0
  B             C             3.0
  B             D             0.0
  C             D             1.0

请注意，我在示例中给出了多种不同的可能性，以确保我的问题清晰。例如，即使对于 Batch_ID=2，B-C 组合出现两次，计数器也不会增加两次。

实现这一目标最有效的方法是什么？

这是我的看法：

from itertools import combinations

def combine(batch):
    """Combine all products within one batch into pairs"""
    return pd.Series(list(combinations(set(batch), 2)))

edges = df.groupby('Batch_ID')['Product_ID'].apply(combine).value_counts()
edges
#(B, C)    3
#(A, B)    1
#(A, C)    1
#(D, C)    1

据我了解，实际上并不需要 0 次出现的边。

如果需要，您可以将索引进一步拆分为源索引和目标索引：

edges = edges.reset_index()
edges = pd.concat([edges, edges['index'].apply(pd.Series)], axis=1)
edges.drop(['index'], axis=1, inplace=True)
edges.columns = 'Weight','Source','Target'
#       Weight Source Target
#0       3      B      C
#1       1      A      B
#2       1      A      C
#3       1      D      C

Or:

c = ['Source', 'Target']
L = edges.index.values.tolist()
edges = pd.DataFrame(L, columns=c).join(edges.reset_index(drop=True))