import networkx as nx
import matplotlib.pyplot as plt
def get_nodes():
with open("nodes.csv","r") as f:
node = f.read().splitlines()
print(node)
return node
def get_links():
with open("links.csv","r") as f:
links = f.read().splitlines()
print(links)
return links
G = nx.MultiDiGraph()
# node节点数据
for ip in get_nodes():
G.add_node(ip[0])
#link数据生成
edges = []
for link in get_links():
data = link.split(",")
#print(data)
des_ip = data[0]
sour_ip = data[1]
edge = (des_ip,sour_ip)
edges.append(edge)
r = G.add_edges_from(edges)
nx.draw(G, with_labels=True, node_color='y', )
plt.show()
