python - 添加到嵌套列表

Question

我想将具有下一个值的元素添加到最嵌套的列表中，i。即.

列表

list_in = [2, 3, [4, 5, [6, 7], 6], 2, [5, 6]]

程序应该返回

list_out = [2, 3, [4, 5, [6, 7, 8], 6], 2, [5, 6]]

如果有 e。 X。两个相等的巢，我想得到:

list_in = [2, [4], [3]]

list_out = [2, [4, 5], [3, 4]]

怎么做？

Answer 1

递归解决方案:

list_in = [2, 3, [4, 5, [6, 7], 6], 2, [5, 6]]


def get_depths(lst, depths, current_depth=0):
    out = [
        get_depths(v, depths, current_depth + 1) if isinstance(v, list) else v
        for v in lst
    ]
    depths.setdefault(current_depth, []).append(out)
    return out


depths = {}
list_out = get_depths(list_in, depths)
for lst in depths[max(depths)]:
    lst.append(lst[-1] + 1)

print(list_out)

打印:

[2, 3, [4, 5, [6, 7, 8], 6], 2, [5, 6]]

---

对于 list_in = [2, [4], [3]] 它打印:

[2, [4, 5], [3, 4]]