python - 将列表部分展平到给定级别

正如评论中指出的那样，对于this相关问题的大多数（如果不是全部）答案对于以下列表均无效：

ls = [1,2,[3,4]]

而且，该列表可以嵌套得更深。如何部分展平到用户给定级别（默认为无穷大）

ls2 = [1,[2,3],[4,[5,6]]]

ls2的所需输出：

展平至第1级：

 [1,2,3,4,[5,6]]

展平至2级（或更高）

 [1,2,3,4,5,6]

最佳答案

您可以递归地执行此操作：

def flatten(l, level=None):
    if level == 0:
        return l
    flattened = []
    for item in l:
        if isinstance(item, list):
            flattened.extend(flatten(item, level-1 if level is not None else None))
        else:
            flattened.append(item)

    return flattened

ls2 = [1,[2,3],[4,[5,6]]]

print(flatten(ls2, level=1))
# [1, 2, 3, 4, [5, 6]]

print(flatten(ls2, level=2))
# [1, 2, 3, 4, 5, 6]

print(flatten(ls2))
# [1, 2, 3, 4, 5, 6]