将csv edgelist读入networkx

时间:2013-09-17 18:41:03

标签: python python-2.7 networkx

我有一个格式为的数据集:

1,2

2,3

1,3

等。 (每对代表两个节点之间的边缘,例如'1,2'是节点1和节点2之间的边缘)

我需要将其读入networkx。目前我正在尝试将其作为列表读取,其中每对是列表中的一个元素,但这不起作用。

1 个答案:

答案 0 :(得分:4)

您可以使用networkx.read_edgelist(file,delimeter =',')。

e.g。

import StringIO
import networkx as nx
data = StringIO.StringIO("""1,2

2,3

1,3
""")

G = nx.read_edgelist(data, delimiter=',', nodetype=str)
for e in G.edges():
    print e
# ('1', '3')
# ('1', '2')
# ('3', '2')