Networkx 数据集创建

Question

我有一个有向图的数据，格式是这样的：

Node1 Node2
A       B
A       C
C       A
D       A

这意味着有一个从A指向B的边，另一个从A指向C，依此类推。我想把这些数据转换成Networkx支持的数据集，以便在Python中进行建模。我希望文件的大小尽可能小，因为我有大约100万个边/链接。

Answer 1

import networkx as nx
import matplotlib.pyplot as plt

content = '''\
Node1 Node2
A       B
A       C
C       A
D       A'''

lines = content.splitlines()
G = nx.DiGraph()
for line in lines[1:]: # skip the first (header) line
    a, b = map(str.strip, line.split())
    G.add_edge(a, b)

nx.draw(G)
plt.savefig('/tmp/graph.png')
# plt.show()