怎么写python3展开列表的代码

这篇文章给大家分享的是有关怎么写python3展开列表的代码的内容。小编觉得挺实用的，因此分享给大家做个参考。一起跟随小编过来看看吧。

展开列表

该方法将通过递归的方式将列表的嵌套展开为单个列表。

def spread(arg):
    ret = []
    for i in arg:
        if isinstance(i, list):
            ret.extend(i)
        else:
            ret.append(i)
    return ret
 
def deep_flatten(lst):
    result = []
    result.extend(
        spread(list(map(lambda x: deep_flatten(x) if type(x) == list else x, lst))))
    return result
 
 
deep_flatten([1, [2], [[3], 4], 5]) # [1,2,3,4,5]

拓展：列表的差

该方法将返回第一个列表的元素，其不在第二个列表内。如果同时要反馈第二个列表独有的元素，还需要加一set_b.difference(set_a)。

def difference(a, b):
    set_a = set(a)
    set_b = set(b)
    comparison = set_a.difference(set_b)
    return list(comparison)
 
 
difference([1,2,3], [1,2,4]) # [3]

感谢各位的阅读！关于怎么写python3展开列表的代码就分享到这里了，希望以上内容可以对大家有一定的帮助，让大家可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到吧！