mirror of
https://github.com/gsi-upm/soil
synced 2024-09-21 07:41:43 +00:00
23 lines
541 B
Python
23 lines
541 B
Python
from networkx import Graph
|
|
import random
|
|
import networkx as nx
|
|
|
|
|
|
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 = random.choice(nodes)
|
|
nodes.remove(n_in) # Avoid loops
|
|
n_out = random.choice(nodes)
|
|
G.add_edge(n_in, n_out)
|
|
return G
|