读取csv数据,然后利用Networkx 做简单的可视化
import networkx as nx
import csv
from itertools import islice
import matplotlib.pyplot as plt
G = nx.Graph()
with open('topology.csv', newline='') as f:
read = csv.reader(f)
for line in islice(read, 1, None):
G.add_edge(line[0], line[1])
nx.draw_networkx(G, pos=nx.spring_layout(G), node_size=10, alpha=0.1, with_labels=None, node_color='#A0CBE2')
plt.show()
结果如下
注意 newline的关键字,可以跳过空行;islice(read, 1, None)则跳过第一行字段名。