pandas でネットワーク解析のための組み合わせを展開したい

Question

以下のようなデータフレームがあります。

id	name
1	a
1	b
1	c
2	d
2	e

これをグラフ解析のために以下のようなデータフレームに変換したいです。

id	name1	name2
1	a	b
1	a	c
1	b	c
2	d	e

よろしくおねがいいたします。

Answer 1

DataFrame.groupby()にてitertools.combinations を適用するとこんな感じで１行で書くどともできます。

import pandas as pd
from itertools import combinations
df = pd.DataFrame({'id':[1,1,1,2,2],'name':['a','b','c','d','e']})

r = df.groupby('id')['name'].apply(lambda d: pd.DataFrame(combinations(d,2))).reset_index(1, drop=True).add_prefix('name')
#   name0 name1
#id
#1      a     b
#1      a     c
#1      b     c
#2      d     e

Answer 2

地道にループでidごとにcombinationsをとればとりあえずできます。

import pandas as pd
from itertools import combinations

df = pd.DataFrame({'id':[1,1,1,2,2],'name':['a','b','c','d','e']})

dfr = pd.DataFrame(columns=['id','name1','name2'])
for id, g in df.groupby('id'):
    for c in combinations(g['name'], 2):
       dfr = dfr.append( pd.Series([id, c[0], c[1]], index=dfr.columns), ignore_index=True)
print(dfr)
"""
  id name1 name2
0  1     a     b
1  1     a     c
2  1     b     c
3  2     d     e
"""

Comments

[退会済みユーザー]

時間がかかりそうですが、確かにできそうですね。

ありがとうございます。