Python 实现 flatten()

永恒的蓝色梦想

Python 实现 flatten()

语法

1. flatten(iterable, /)

复制代码

参数

参数	描述
iterable	可迭代对象

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

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

实现方式
递归

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

例子

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

复制代码

1. >>> def gen1():
   
2.         yield 1
   
3.         yield 2
   
4.         yield 3
   
5. 
   
6. >>> def gen2():
   
7.         yield 4
   
8.         yield gen1()
   
9. 
   
10. >>> def gen3():
    
11.         yield 5
    
12.         yield 6
    
13.         yield gen2()
    
14. 
    
15. >>> def gen4():
    
16.         yield 7
    
17.         yield gen3()
    
18. 
    
19. >>> list(gen4())
    
20. [7, <generator object gen3 at 0x000001EF0093B5F0>]
    
21. >>> list(flatten(gen4()))
    
22. [7, 5, 6, 4, 1, 2, 3]

复制代码

代码

1. def flatten(it, /):
   
2.     for i in it:
   
3.         if hasattr(i,'__iter__'):
   
4.             yield from flatten(i)
   
5.         
   
6.         else:
   
7.             yield i

复制代码

永恒的蓝色梦想

二楼