Skip to main content
`Ctrl`+`K`

Note

Go to the end to download the full example code.

# Multipartite Layout

    import itertools
    import matplotlib.pyplot as plt
    import networkx as nx

    subset_sizes = [5, 5, 4, 3, 2, 4, 4, 3]
    subset_color = [
        "gold",
        "violet",
        "violet",
        "violet",
        "violet",
        "limegreen",
        "limegreen",
        "darkorange",
    ]

    def multilayered_graph(*subset_sizes):
        extents = nx.utils.pairwise(itertools.accumulate((0,) + subset_sizes))
        layers = [range(start, end) for start, end in extents]
        G = nx.Graph()
        for i, layer in enumerate(layers):
            G.add_nodes_from(layer, layer=i)
        for layer1, layer2 in nx.utils.pairwise(layers):
            G.add_edges_from(itertools.product(layer1, layer2))
        return G

    G = multilayered_graph(*subset_sizes)
    color = [subset_color[data["layer"]] for v, data in G.nodes(data=True)]
    pos = nx.multipartite_layout(G, subset_key="layer")
    plt.figure(figsize=(8, 8))
    nx.draw(G, pos, node_color=color, with_labels=False)
    plt.axis("equal")
    plt.show()

Total running time of the script: (0 minutes 0.056 seconds)

Gallery generated by Sphinx-Gallery