Networkx数据集创建

Question

我有

形式的有向图数据

Node1 Node2
A       B
A       C
C       A
D       A

这意味着，A - >之间的有向边/链接; B，A - ＆gt; C等等。 我想将这些数据创建到Networkx支持的数据集中，以便在Python中进行建模。我希望尽可能减小文件大小。我有大约1百万的边缘/链接。

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