for i, o in enumerate(tree):
    if i == 0:
        continue
if idx[0] == 0:
    return tree, i
idx[0] -= 1
if isinstance(o, list):
    container, result_index = find_idx(o, idx)
if container is not None:
    return container, result_index
return None, None
Как разделить два вложенных списка и объединить части для создания двух новых вложенных списков