Mercurial > repos > shellac > sam_consensus_v3
view env/share/doc/networkx-2.5/examples/drawing/plot_multipartite_graph.py @ 0:4f3585e2f14b draft default tip
"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"
author | shellac |
---|---|
date | Mon, 22 Mar 2021 18:12:50 +0000 |
parents | |
children |
line wrap: on
line source
""" =================== Multipartite Layout =================== """ import itertools import matplotlib.pyplot as plt import networkx as nx from networkx.utils import pairwise 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 = 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 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()