mirror of
https://github.com/gsi-upm/soil
synced 2024-12-23 00:28:11 +00:00
27 lines
555 B
Python
27 lines
555 B
Python
|
from networkx import Graph
|
||
|
|
import networkx as nx
|
||
|
|
from random import choice
|
||
|
|
|
||
|
|
def mygenerator(n=5, n_edges=5):
|
||
|
|
'''
|
||
|
|
Just a simple generator that creates a network with n nodes and
|
||
|
|
n_edges edges. Edges are assigned randomly, only avoiding self loops.
|
||
|
|
'''
|
||
|
|
G = nx.Graph()
|
||
|
|
|
||
|
|
for i in range(n):
|
||
|
|
G.add_node(i)
|
||
|
|
|
||
|
|
for i in range(n_edges):
|
||
|
|
nodes = list(G.nodes)
|
||
|
|
n_in = choice(nodes)
|
||
|
|
nodes.remove(n_in) # Avoid loops
|
||
|
|
n_out = choice(nodes)
|
||
|
|
G.add_edge(n_in, n_out)
|
||
|
|
return G
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|