python展开嵌套list,flatten 嵌套 list

比如有一个list,嵌套了三层:

a = [[1, 2], ['a', 'b', [3, 4]]]

现在想将其变为:

a = [1, 2, '1', 'd', 3, 4]

也就是说,无论嵌套了多少层的list,我们都想将其flatten为一层的list。

解决方案:

实现一个递归函数:

def flatten(nest_list:list):
    out = []
    for i in nest_list:
        if isinstance(i, list):
            tmp_list = flatten(i)
            for j in tmp_list:
                out.append(j)
        else:
            out.append(i)
    return out

其实可以更加简化:

def flatten(nest_list:list):
    return [j for i in nest_list for j in flatten(i)] if isinstance(nest_list, list) else [nest_list]

测试输出:


a = [[1,2], ['a','b', [3, 4]]]
b = flatten(a)
print(b)
# 输出: [1, 2, 'a', 'b', 3, 4]

好用的话,记得来点赞。

你可能感兴趣的:(python,python,大数据,开发语言)