python 平展 语法糖演示

flatten.py

b = [[1, 2], [3], [[1, [2]]]]

flatten1 = lambda x: [y for l in x for y in flatten1(l)] if type(x) is list else [x]


def flatten2(x):
    res = []
    if type(x) is list:
        for l in x:
            for y in flatten2(l):
                res.append(y)
    else:
        res = [x]
    return res


print(flatten1(b))
print(flatten2(b))

if __name__ == '__main__':
    pass