我正在使用python和networkx将边缘随机分配给点头。点头分为三类(白色,黑色和其他),每个类别有33个节点。代码正在运行,但我有两个问题: 1-如何确保一个节点不会被选中两次?我的意思是让我们说在第一轮中,在节点4和56之间定义了一个边缘。我怎样才能确保在第4轮中这个边缘不会被再次选择? 2-我想要做的下一步是分配怀特。这意味着例如,如果x是白色,则y是A%,更可能是白色。我该如何添加呢?
import networkx as nx
import matplotlib.pyplot as plt
import random
import numpy
G=nx.Graph()
w=1
b=34
o=67
while w < 34:
G.add_node(w, race='white')
w+=1
while b < 67:
G.add_node(b, race='black')
b+=1
while o < 100:
G.add_node(o, race='other')
o+=1
from numpy import random as rand
###first round edges assignment
num1edge = int(raw_input("Please enter number of edges you want to start with: "))
i=0
while i< num1edge:
x1 = rand.randint (1, 99)
y1 = rand.randint (x1, 99)
G.add_edge(x1,y1)
i+=1
numrounds = int(raw_input("Please enter how many times you want to run: "))
numedge = int(raw_input("Please enter number of edges you want to be created in each round: "))
j = 0
k = 0
while j < numrounds:
while k < num1edge:
x = rand.randint (1, 99)
y = rand.randint (x, 99)
G.add_edge(x,y)
k+=1
j+=1
nx.draw(G)
plt.show()
答案 0 :(得分:2)
使用邻接矩阵。行和列之间的拦截为您提供了点头之间的关系。例如,假设您只有3个节点。 1,2和3,所以如果你有下一个矩阵
1 2 3
_________
1 | 0 0 0
2 | 0 0 1
3 | 0 0 0
表示已选择节点2和3之间的边。如果你选择另一个,让我们说(1,2),只需更新你的矩阵:
your_matrix[1][2] = 1