我想在Python中使用字典模拟图形。
G = {'a':{'b':3, 'c':4},
'b':{'a':3, 'c':5},
'c':{'a':4,'b':5},
}
G是一个字典,其中每个值都是一个字典,代表相邻节点及其权重。
我有以下代码:
def Graph(nodes):
list_nodes = []
list_adjacent_nodes = []
graph = {}
for i in range(nodes):
node = raw_input(" ID Node %d:" %(i+1))
list_nodes.append(node)
num_adjacent_nodes = input(" Num adjacent nodes '%s':" %(list_nodes[i]))
list_adjacent_nodes.append([])
for j in range(num_adjacent_nodes):
adjacent_node = raw_input(" ID adjacent nodes %d:" %(j+1))
list_adjacent_nodes.append[i].append(adjacent_node)
weight = input(" Wight (Nodes(%s,%s) ): " %(list_nodes[i],list_adjacent_nodes[i][j] ) )
graph[list_nodes[i]] = {}
graph[lista_nodes[i]][list_adjacent_nodes[i][j]] = weight
return graph
但是,我得到以下结果:
G = {'a':{'c':4}, 'b':{'c':5}, 'c':{'b':5}, }
我哪里出了问题,如何退回所需的输出?
答案 0 :(得分:2)
你覆盖"内部"循环的每次迭代中的字典,带有空字典。只有在缺失时才应该这样做:
if not graph[list_nodes[i]]: # Here!
graph[list_nodes[i]] = {}
graph[list_nodes[i]][list_adjacent_nodes[i][j]] = weight