Python:循环查找每个node_id的邻居

时间:2018-02-02 15:24:45

标签: python python-3.x algorithm loops shortest-path

我有一个如下所示的循环:

i = 0
while i < len(node_id):
    for arc in self._adjacency_lists[i]:
        neighbours.append(arc.head_node_id)
        i += 1

给了我输出:

[1, 2, 2, 3, 1, 3]

其中第0个和第1个元素是node_id = 0的邻居,2个node_id的邻居= 1,3个node_id = 1的3个,node_id = 3的1个,node_id的2个= 4个。如何追加这些邻居到一个新的列表,所以我的输出看起来像:

new_list = [[1,2], [2], [3], [1], [3]]

1 个答案:

答案 0 :(得分:0)

您应首先为每个节点附加一个空列表,然后将邻居添加到该节点。

{{1}}