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
|
{
"cells": [
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"collapsed": false
},
"outputs": [],
"source": [
"%matplotlib inline"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"\n# Multipartite Layout\n"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"collapsed": false
},
"outputs": [],
"source": [
"import itertools\nimport matplotlib.pyplot as plt\nimport networkx as nx\n\nsubset_sizes = [5, 5, 4, 3, 2, 4, 4, 3]\nsubset_color = [\n \"gold\",\n \"violet\",\n \"violet\",\n \"violet\",\n \"violet\",\n \"limegreen\",\n \"limegreen\",\n \"darkorange\",\n]\n\n\ndef multilayered_graph(*subset_sizes):\n extents = nx.utils.pairwise(itertools.accumulate((0,) + subset_sizes))\n layers = [range(start, end) for start, end in extents]\n G = nx.Graph()\n for (i, layer) in enumerate(layers):\n G.add_nodes_from(layer, layer=i)\n for layer1, layer2 in nx.utils.pairwise(layers):\n G.add_edges_from(itertools.product(layer1, layer2))\n return G\n\n\nG = multilayered_graph(*subset_sizes)\ncolor = [subset_color[data[\"layer\"]] for v, data in G.nodes(data=True)]\npos = nx.multipartite_layout(G, subset_key=\"layer\")\nplt.figure(figsize=(8, 8))\nnx.draw(G, pos, node_color=color, with_labels=False)\nplt.axis(\"equal\")\nplt.show()"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.9.16"
}
},
"nbformat": 4,
"nbformat_minor": 0
}
|