Python压平嵌套列表的一种方法

版权声明:本博文欢迎分享与转载,转载请注明出处和作者。 https://blog.csdn.net/dream6104/article/details/88742226

把嵌套级别的多级列表看成是树状结构:

定义函数:

def flat(tree):
    res = []
    for i in tree:
        if isinstance(i, list):
            res.extend(flat(i))
        else:
            res.append(i)
    return res

z=[[1,2],[3,4],[[5,6],[7,8]]]


flatten(z)输出

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

猜你喜欢

转载自blog.csdn.net/dream6104/article/details/88742226