python - 从 Pandas 中的列生成对

Question

Pandas 中最简单的方法是什么

df = pd.DataFrame({'Class': [1, 2], 'Students': ['A,B,C,D', 'E,A,C']})
df

   Class Students
0      1  A,B,C,D
1      2    E,A,C

进入这个？

Answer 1

让我们试试组合:

from functools import partial
from itertools import combinations

(df.set_index('Class')['Students']
   .str.split(',')
   .map(partial(combinations, r=2))
   .map(list)
   .explode()
   .reset_index())

   Class Students
0      1   (A, B)
1      1   (A, C)
2      1   (A, D)
3      1   (B, C)
4      1   (B, D)
5      1   (C, D)
6      2   (E, A)
7      2   (E, C)
8      2   (A, C)