-
Notifications
You must be signed in to change notification settings - Fork 0
/
visualize.py
62 lines (38 loc) · 1.45 KB
/
visualize.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# Importing dependencies
import networkx as nx
import numpy as np
import matplotlib
matplotlib.use('Agg')
from matplotlib import pyplot as plt
import math
def build_graph(n, adjMatrix, location):
# Receieving number of vertices, adjacency matrix and location of where image has to be stored
# To generate graph with random vertices layout
G = nx.lollipop_graph(n, 0)
# Creating a directed graph
G = nx.DiGraph()
"""#Relabelling node names to alphabets
mapping={}
char=65
for i in range(n):
mapping[i]=chr(char)
char+=1"""
#adjMatrix = np.array([[math.inf, -5, 2, 3], [math.inf, math.inf, 4, math.inf], [math.inf, math.inf, math.inf, 1], [math.inf, math.inf, math.inf, math.inf]])
length = len(adjMatrix)
# Adding edges to graph
for i in range(length):
for j in range(length):
if(adjMatrix[i,j] == math.inf):
continue
G.add_edge(i, j, weight=adjMatrix[i, j])
#print(adjMatrix[i, j])
# Retrieving position of nodes
#pos = nx.random_layout(G)
pos=nx.circular_layout(G)
# Drawing the finalised graph
nx.draw(G, pos, with_labels=True, font_weight='bold')
labels = nx.get_edge_attributes(G, 'weight')
nx.draw_networkx_edge_labels(G, pos, edge_labels=labels)
plt.savefig(location, format="PNG")
# plt.show()
plt.clf()