python从嵌套列表父母元素中查找值

Question

我用python保存了一个父母的结构。

PC_List = [[1,1,2],[1,2,8],[1,3,5],[2,4,7],[3,5,1]]

子列表元素的第一个索引是parentid， 子列表元素的第二个索引是childid， 子列表元素的第三个索引是值，

什么是从childid =当前parentid的子列表元素附加值的最佳和最快的方法？ 新列表应该看起来像这样，

New_PC_List = [[1,1,2],[1,2,8,2],[1,3,5,2],[2,4,7,8],[3,5,1,5]]

我会很感激一些tipps。 PC_LIST的PS Len可以超过100万行。

谢谢

Answer 1

我首先要构造一个尺寸为“ value”的映射“ id”，然后使用它来附加值。

PC_List = [[1,1,2],[1,2,8],[1,3,5],[2,4,7],[3,5,1]]
values = {k: v for _, k, v in PC_List}

output = [[p, c, v, values[p]] for p, c, v in PC_List]
print(output)
# [[1, 1, 2, 2], [1, 2, 8, 2], [1, 3, 5, 2], [2, 4, 7, 8], [3, 5, 1, 5]]

Answer 2

很难遵循您想要的东西，但是在将列表的第一个元素列为0的父元素ID之后（您使用1个索引，所以我认为没关系），我认为这可以做到您想要的：

PC_List = [[0,1,2],[1,2,8],[1,3,5],[2,4,7],[3,5,1]]
for i in range(1,len(PC_List)):
    PC_List[i].append(PC_List[PC_List[i][0]-1][2])

此解决方案更改原始列表，但可以使用copy.deepcopy修复。