将邻接矩阵转邻接表时字典仅返回最后一项，如何获取完整条目？

你的代码只返回最后一项，核心原因是每次循环都重新创建了adjlist字典，之前循环生成的键值对都会被覆盖，最后只剩最后一次循环的结果。

修复方案 #

1. 先在循环外初始化一个空字典，避免每次循环覆盖原有数据
2. 内层循环处理完当前节点的邻接列表后，直接将键值对添加到已有的字典中

修正后的完整代码 #

from typing import List, Dict

def adjmat_to_adjlist(adjmat: List[List[int]]) -> Dict[int, List[int]]:
    adjlist = {}  # 初始化空字典，放在循环外部
    for key, row in enumerate(adjmat, start=1):
        neighbors = []   
        for node, value in enumerate(row, start=1):
            if value != 0:
                neighbors.append(node)
        adjlist[key] = neighbors  # 向字典添加当前节点的邻接列表
    return adjlist

示例验证 #

比如输入如下邻接矩阵：

test_mat = [
    [0, 1, 0],
    [1, 0, 1],
    [0, 1, 0]
]
print(adjmat_to_adjlist(test_mat))

输出结果会是完整的邻接表：

{1: [2], 2: [1, 3], 3: [2]}

内容的提问来源于stack exchange，提问作者guidedisp