import networkx as nx
import itertools
import random
def setup_nodes_edges(n, p_matrix):
nodes = list(range(n))
edges = set()
for combination in itertools.combinations(nodes, 2):
x, y = combination
if p_matrix[x][y] <= random.random():
edges.add(combination)
return nodes, edges
nodes, edges = setup_nodes_edges(3, p_matrix) # I assume you have a p_matrix
G = nx.Graph()
G.add_nodes_from(nodes)
G.add_edges_from(edges)
import networkx as nx
import matplotlib.pyplot as plt
import numpy as np
N = 8
m = np.random.random(N*N).reshape(N, N)
for i in range(N): m[i, i] = 0
nodes = range(N)
edges = {}
for i in range(N):
for j in range(N):
if np.random.random() < m[i, j]:
edges[(i, j)] = round(m[i, j], 2)
G = nx.Graph()
G.add_nodes_from(nodes)
G.add_edges_from(edges)
pos = nx.spring_layout(G)
nx.draw(G, pos)
nx.draw_networkx_edge_labels(G, pos, edge_labels=edges)
plt.show()
#!/usr/bin/env python
import numpy as np
import networkx as nx
N = 10 # number of nodes
P = np.random.rand(10, 10) # your "matrix of probabilities"
adjacency = np.random.rand(*P.shape) <= P # adjacency[ii, jj] is True with probability P[ii, jj]
graph = nx.from_numpy_matrix(adjacency, nx.DiGraph) # assuming the graph is supposed to be directed, presumably if P[ii, jj] != P[jj, ii]
我认为您只需要根据需要设置节点和边,并将其传递给图形构造函数
例如—
result
在这种情况下,很容易推出自己的图形生成器
相关问题 更多 >
编程相关推荐