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