python - 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()