Python 实现 flatten()

永恒的蓝色梦想

Python 实现 flatten()

语法

flatten(iterable, /)

参数

参数	描述
iterable	可迭代对象

描述
flatten 函数可以用来压平可迭代对象。

返回值
这个函数会返回一个生成器。

实现方式
递归

原理
如果 iterable 中的一个元素实现了 __iter__ 方法，则递归取它的值。

例子

>>> [1,[2,[3,[4,[5,[6,[7]]]]]]]
[1, [2, [3, [4, [5, [6, [7]]]]]]]
>>> list(flatten(_))
[1, 2, 3, 4, 5, 6, 7]

>>> def gen1():
        yield 1
        yield 2
        yield 3

>>> def gen2():
        yield 4
        yield gen1()

>>> def gen3():
        yield 5
        yield 6
        yield gen2()

>>> def gen4():
        yield 7
        yield gen3()

>>> list(gen4())
[7, <generator object gen3 at 0x000001EF0093B5F0>]
>>> list(flatten(gen4()))
[7, 5, 6, 4, 1, 2, 3]

代码

def flatten(it, /):
    for i in it:
        if hasattr(i,'__iter__'):
            yield from flatten(i)
        
        else:
            yield i

永恒的蓝色梦想

二楼