跳转至

maxcut_draw

QuICT.tools.display.graph_drawer 

graph_drawer(nodes, edges, title: str = 'Graph', save_path=None)

Draw an undirected graph based on given nodes and edges.

Parameters:

  • nodes (list) –

    The nodes of the graph.

  • edges (list) –

    The edges of the graph.

  • title (str, default: 'Graph' ) –

    The title of the figure. Defaults to "Graph".

  • save_path (str, default: None ) –

    The path to save the figure. Defaults to None.

Source code in QuICT/tools/display/maxcut_draw.py 
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
def graph_drawer(nodes, edges, title: str = "Graph", save_path=None):
    """Draw an undirected graph based on given nodes and edges.

    Args:
        nodes (list): The nodes of the graph.
        edges (list): The edges of the graph.
        title (str, optional): The title of the figure. Defaults to "Graph".
        save_path (str, optional): The path to save the figure. Defaults to None.
    """
    plt.figure()
    plt.title(title)
    G = nx.Graph()
    G.add_nodes_from(nodes)
    G.add_edges_from(edges)
    pos = nx.circular_layout(G)
    nx.draw_networkx(G, pos, node_color="#416DB6", **OPTIONS)
    ax = plt.gca()
    ax.margins(0.20)
    plt.axis("off")
    if save_path:
        plt.savefig(save_path + "/{}.png".format(title), transparent=True)
    plt.show()
    return

QuICT.tools.display.result_drawer 

result_drawer(nodes, edges, solution_bit, title: str = 'The result of MaxCut', save_path=None)

Draw an undirected graph based on given nodes and edges.

Parameters:

  • nodes (list) –

    The nodes of the graph.

  • edges (list) –

    The edges of the graph.

  • solution_bit (str) –

    The result state as a binary string.

  • title (str, default: 'The result of MaxCut' ) –

    The title of the figure. Defaults to "The result of MaxCut".

  • save_path (str, default: None ) –

    The path to save the figure. Defaults to None.

Source code in QuICT/tools/display/maxcut_draw.py 
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
def result_drawer(
    nodes,
    edges,
    solution_bit,
    title: str = "The result of MaxCut",
    save_path=None,
):
    """Draw an undirected graph based on given nodes and edges.

    Args:
        nodes (list): The nodes of the graph.
        edges (list): The edges of the graph.
        solution_bit (str): The result state as a binary string.
        title (str, optional): The title of the figure. Defaults to "The result of MaxCut".
        save_path (str, optional): The path to save the figure. Defaults to None.
    """
    plt.figure()
    plt.title(title)
    G = nx.Graph()
    G.add_nodes_from(nodes)
    G.add_edges_from(edges)
    pos = nx.circular_layout(G)

    node_color = ["red" if solution_bit[v] == "1" else "#416DB6" for v in nodes]
    edge_color = []
    edge_style = []
    for u, v in G.edges:
        if solution_bit[u] != solution_bit[v]:
            edge_color.append("red")
            edge_style.append("dashed")
        else:
            edge_color.append("black")
            edge_style.append("solid")

    nx.draw(
        G,
        pos,
        node_color=node_color,
        edge_color=edge_color,
        style=edge_style,
        **OPTIONS
    )
    ax = plt.gca()
    ax.margins(0.20)
    plt.axis("off")
    if save_path:
        if not os.path.exists(save_path):
            os.makedirs(save_path)
        plt.savefig(save_path + "/{}.png".format(title), transparent=True)
    plt.show()