你知道如何将python中嵌套的列表扁平化吗?
楔子
将python中嵌套的列表扁平化,指的就是:
lst = [1, [[[[3, 3], 5]]], [[[[[[[[[[[[[6]]]]], 8]]], "aaa"]]]], 250]]
# 类似于上面的列表,我们要得到下面的结果
[1, 3, 3, 5, 6, 8, "aaa", 250]
那么我们要如何做呢?
yield和yield from
关于如何把一个嵌套的列表扁平化,我们需要使用yield和yield from。这两位老铁的底层实现原理我不想深究,因为牵扯的东西太多了,况且针对目前的问题我们也用不到,这里我们简单的说一下。
def foo():
yield [1, 2, 3]
def bar():
yield from [1, 2, 3]
print(foo().__next__()) # [1, 2, 3]
print(bar().__next__()) # 1
"""
我们看到关于yield,它是把yield后面的整个元素都迭代出来
但是对于yield from,它的后面必须跟着一个可迭代的对象,然后依次迭代这个可迭代对象里面的每一个元素
"""
因此,我们就找到了方法。
def flat(l: list):
for _ in l:
if isinstance(_, list):
yield from flat(_)
else:
yield _
lst = [1, [[[[3, 3], 5]]], [[[[[[[[[[[[[6]]]]], 8]]], "aaa"]]]], 250]]
print([_ for _ in flat(lst)]) # [1, 3, 3, 5, 6, 8, 'aaa', 250]
如果觉得文章对您有所帮助,可以请囊中羞涩的作者喝杯柠檬水,万分感谢,愿每一个来到这里的人都生活愉快,幸福美满。
微信赞赏
支付宝赞赏