-
Notifications
You must be signed in to change notification settings - Fork 0
/
plot_gate_map.py
37 lines (34 loc) · 1.13 KB
/
plot_gate_map.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
import pandas as pd
import networkx as nx
from matplotlib import pyplot as plt
import os
os.environ["QT_QPA_PLATFORM"] = "xcb"
G = nx.DiGraph()
G.add_edges_from(
[('e1', 'q1'), ('e1', 'q2'), ('e1', 'q3'), ('e1', 'e2'),
('e2', 'q4'), ('e2', 'q5'), ('e2', 'q6'), ("e2", "e3"),
('e3', 'q7'), ('e3', 'q8'), ('e3', 'q9'), ('e3', 'e2') ])
color_map = []
for node in G:
if node.find("e") != -1:
color_map.append('blue')
else:
color_map.append('red')
pos = nx.layout.shell_layout(G)
nx.draw(G, with_labels=True, node_size=500, alpha=0.95, linewidths=15, node_color=color_map, pos=pos)
plt.show()
G = nx.DiGraph()
G.add_edges_from(
[('000', '000'), ('000', '001'), ('000', '010'), ('000', '100'),
('111', '111'), ('111', '110'), ('111', '101'), ("111", "011")])
color_map = []
for node in G:
if(not node.find("000")):
color_map.append('green')
elif(not node.find("111")):
color_map.append('green')
else:
color_map.append('red')
pos = nx.layout.spring_layout(G)
nx.draw(G, with_labels=True, node_size=500, alpha=0.95, linewidths=15, node_color=color_map, pos=pos)
plt.show()