无法传输邻接列表数据并将其转换为邻接矩阵

时间:2017-11-19 17:52:35

标签: python algorithm list matrix

我的代码首先接受用户输入:

  • n =图表中的顶点数

  • file =要计算的文件

在这种情况下,我的文件是一个包含以下内容的.txt文件:

0 1  
2 1  
0 2  
1 3 

我的代码成功将其转换为输出[[1, 2], [0, 2, 3], [1, 0], [1]]的邻接列表。然而,问题是将其转换为邻接矩阵。我知道我的代码的主要问题是在这一行:

for x in range(len(vertices)):
    matrix[z-1][vertices[x]] = 1

这是我的完整代码:

n = int(input("Enter the number of vertices: ")) ## E.g. 4
file = input("Enter the filename: ") ## E.g. graph.txt

vertices = []
matrix = [] ## define list

for x in range(n):
    matrix.append([0]*n]) ## append a list for n, e.g. if n = 4 then [[][][][]]
    vertices.append([])

f =open(file)

z = 0
for line in f: ##loop statement until no more line in file
    line = line.split() ## turn every line into a list
    z+=1



    for y in range(len(line)):
        line[y] = line[y].strip() ## remove spaces
        line[y] = int(line[y]) ## converts line list into integer
           ## add  value of line into main list

    vertices[line[0]].append(line[1])
    vertices[line[1]].append(line[0])



for z in range(len(matrix)):

    for x in range(len(vertices)):
        matrix[z-1][vertices[x]] = 1



print(vertices)
print(matrix)

1 个答案:

答案 0 :(得分:0)

首先,您使用z-1作为索引,可以为-1,因此应该只是z

其次,您在内部循环中访问vertices[x],就好像它有一个节点号一样,但它是一个嵌套列表。相反,您需要vertices[z][x]。为了提高性能,原则上最好为vertices[z]创建一个名称:

for z in range(len(matrix)):
    neighbors = vertices[z]
    for x in range(len(neighbors)):
        matrix[z][neighbors[x]] = 1

或者:

for z, neighbors in enumerate(vertices):
    for x in neighbors:
        matrix[z][x] = 1