Networkx draw tripartite graph

import networkx as nx
import numpy as np

BG = nx.Graph()
source = ['s']
first = np.arange(3)
second = np.arange(3, 8)

BG.add_nodes_from(source, bipartite=0)
BG.add_nodes_from(first, bipartite=1)
BG.add_nodes_from(second, bipartite=2)
source_first_edges = []
first_second_edges = []

for f in first:
    source_first_edges.append(('s', f))
for s in second:
    for f in first:
        first_second_edges.append((f, s))

BG.add_edges_from(source_first_edges)
BG.add_edges_from(first_second_edges)

nodes = BG.nodes()
# for each of the parts create a set 
nodes_0  = set([n for n in nodes if  BG.nodes[n]['bipartite']==0])
nodes_1  = set([n for n in nodes if  BG.nodes[n]['bipartite']==1])
nodes_2  = set([n for n in nodes if  BG.nodes[n]['bipartite']==2])

# set the location of the nodes for each set
pos = dict()
pos.update( (n, (1, y)) for y, n in enumerate(nodes_0) ) # put nodes from X at x=1
pos.update( (n, (2, y)) for y, n in enumerate(nodes_1) ) # put nodes from Y at x=2
pos.update( (n, (3, y)) for y, n in enumerate(nodes_2) ) # put nodes from X at x=1
pos.update( (n, (4, y)) for y, n in enumerate(nodes_3) )

nx.draw_networkx(BG, pos=pos,)

Source: stackoverflow.com

Similar codes

0 vote216 views1 answer
Strptime() many format
Python
7 votes216 views2 answers
Your models have changes that are not yet reflected in a migration, and so won't be applied. Run 'manage.py makemigrations' to make new migrations, and then re-run 'manage.py migrate' to apply them.
Python
0 vote171 views1 answer
How to return only fractional part in python
Python
0 vote176 views1 answer
Predict probabilities with xg boost
Python
0 vote167 views1 answer
No default language could be detected for django app
Python
0 vote177 views1 answer
Avoir l'indice d'un nombre dans un array numpy
Python
6 votes163 views1 answer
Selectfield flask wtf
Python
-1 vote215 views2 answers
Python dict remove duplicates where name are not the same
Python
0 vote177 views1 answer
Discord.py autorole
Python
0 vote272 views2 answers
Scikit learn decistion tree
Python