【Python】禁止列表套娃！

ckblt

今天闲得没事干突发奇想做了个禁止列表套娃的函数，
把套娃的列表弄成不套娃的列表，
如：

1. [ [ 1, 2, 3 ],
   
2.   [ 4, 5, 6 ],
   
3.   [ 7, 8, 9 ] ] # 套娃
   
4. [ 1, 2, 3, 4, 5, 6, 7, 8, 9 ] # 不套娃

复制代码

（小插曲：我由于懒得翻译“禁止列表套娃”，直接写成“no_list_list”了 ）

废话少说，上代码！

1. from typing import Union
   
2. 
   
3. 
   
4. def no_list_list(
   
5.     x: Union[list, tuple, set], no_tuples: bool = False, no_sets: bool = False
   
6. ) -> list:
   
7.     """
   
8.     禁止列表套娃函数（使用递归）
   
9.     返回值: 列表
   
10. 
    
11.     ### 参数:
    
12. 
    
13.     x: 套娃的 列表 | 元组 | 集合
    
14.     no_tuples: 禁止元组套娃
    
15.     no_sets: 禁止集合套娃
    
16. 
    
17.     ### 例子:
    
18. 
    
19.     ```
    
20.     no_list_list( [ [ [ 1, 2, 3, 4, 5 ] ] ] )
    
21.     [1, 2, 3, 4, 5]
    
22. 
    
23.     no_list_list( [ [ [ 1, 2, 3, 4, 5 ], 6 ], 7 ] )
    
24.     [1, 2, 3, 4, 5, 6, 7]
    
25.     ```
    
26.     """
    
27. 
    
28.     new_list = []
    
29. 
    
30.     for i in x:
    
31.         if (
    
32.             (isinstance(i, list))
    
33.             or (no_tuples and isinstance(i, tuple))
    
34.             or (no_sets and isinstance(i, set))
    
35.         ):
    
36.             new_list.extend(no_list_list(i, no_tuples, no_sets))
    
37.         else:
    
38.             new_list.append(i)
    
39. 
    
40.     return new_list

复制代码

如有疑问，请回复我哟！

ckblt

青出于蓝 发表于 2022-1-26 13:04
想法很好~语法好像有很多问题......

发现了一个问题（list|tuple|set改成Union[list,tuple,set]）

还有什么问题呢（我用的是Python3.10，没有报错）

青出于蓝

ckblt 发表于 2022-1-26 13:14
发现了一个问题（list|tuple|set改成Union）

还有什么问题呢（我用的是Python3.10，没有报错）

没问题了