Question

尝试编写代码以将图形表示形式从邻接列表转换为邻接矩阵。

该函数应输入一个邻接表adj_list = [[1,2],[2],[],[0,1]]并输出二进制矩阵

adj_mat = [0. 1. 1. 0]
          [0. 0. 1. 0.]
          [0. 0. 0. 0.]
          [1. 1. 0. 0.]

但是运行代码

 def adj_list_to_matrix(adj_list):
    n = len(adj_list)
    adj_matrix = np.nan * np.ones((n,n))
    np.fill_diagonal(adj_matrix,0)

    for i in range(n):
        for j, w in adj_list[i]:
            adj_matrix[i,j] = w
    return adj_matrix

产生错误消息

for j, w in adj_list[i]:
TypeError: cannot unpack non-iterable int object

任何人都可以帮助解决此问题以获得有效的代码吗？

Answer 1

for j, w in adj_list[i]:不起作用，因为adj_list[i]只是一个列表，因此您只能在for循环中解压缩一个值。似乎您希望w是一个加权因子，但是邻接列表没有任何加权。

您可以执行此操作，假设所有权重均为1（我认为这是您根据问题中的预期输出所要的）。

def adj_list_to_matrix(adj_list):
    n = len(adj_list)
    adj_matrix = np.nan * np.ones((n,n))
    np.fill_diagonal(adj_matrix,0)

    for i in range(n):
        for j in adj_list[i]:
            adj_matrix[i,j] = 1
    return adj_matrix

输出

[[ 0.,  1.,  1., nan],
 [nan,  0.,  1., nan],
 [nan, nan,  0., nan],
 [ 1.,  1., nan,  0.]])

加权邻接矩阵

如果您需要加权邻接矩阵，则需要将权重放在adj_list中，如下所示：

adj_list = [{1:0.2,2:0.5},{2:1},{},{0:0.1,1:0.6}]

然后您的代码只需要进行少量调整即可工作

def adj_list_to_matrix(adj_list):
    n = len(adj_list)
    adj_matrix = np.nan * np.ones((n,n))
    np.fill_diagonal(adj_matrix,0)

    for i in range(n):
        for j, w in adj_list[i].items():
            adj_matrix[i,j] = w
    return adj_matrix

输出

[[0. , 0.2, 0.5, nan],
 [nan, 0. , 1. , nan],
 [nan, nan, 0. , nan],
 [0.1, 0.6, nan, 0. ]]

在python中将邻接列表转换为邻接矩阵

1 个答案:

加权邻接矩阵