python - 将 csv edgelist 读入 networkx

Question

我有以下格式的数据集:

1,2

2,3

1,3

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

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

Answer 1

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

例如

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')