部分列表在Python中展平

Question

我有一個小問題，我缺乏一些Python體驗。 假設列表：

list=[[40, 20, 45, 40], [[20, 10, 10, 30, 45, 20], [30, 20, 20, 30]]] 

我想要做的東西，我稱之爲「局部扁平化」，因爲預期的輸出結果是：

[[40, 20, 45, 40], [20, 10, 10, 30, 45, 20], [30, 20, 20, 30]] 

通知之前的list的lentgh是2而預期輸出有長度3。問題是我不知道list中的哪個元素嵌套了更深一層。

This answer和許多其他人沒有什麼幫助，因爲扁平化程度很高。以下問題的輸出是：

list=[40, 20, 45, 40, [20, 10, 10, 30, 45, 20], [30, 20, 20, 30]] 

（注意缺少的括號從列表中的第4個元素

Answer 1

您可以創建一個遞歸函數，測試內部元件是否all是列表如果是這樣。遞歸應用算法，否則產生列表本身

def inner_lists(lst): 
    if all(isinstance(x, list) for x in lst): 
     for inner in lst: 
      for x in inner_lists(inner): 
       yield x 
    else: 
     yield lst 

>>> lst = [[40, 20, 45, 40], [[20, 10, 10, 30, 45, 20], [30, 20, 20, 30]]] 
>>> list(inner_lists(lst)) 
[[40, 20, 45, 40], [20, 10, 10, 30, 45, 20], [30, 20, 20, 30]] 

或者使用yield from（Python的3.3和更高版本）：

def inner_lists(lst): 
    if all(isinstance(x, list) for x in lst): 
     yield from (x for inner in lst for x in inner_lists(inner)) 
    else: 
     yield lst 

臺

或者沒有發電機：

def inner_lists(lst): 
    if all(isinstance(x, list) for x in lst): 
     return [x for inner in lst for x in inner_lists(inner)] 
    else: 
     return [lst] 

Answer 2

如果你的列表中只能包含數字或子列表（而不是兩個都），可以使用以下發生器功能：

In [25]: def partial_flattening(lst): 
      for sub in lst: 
       if isinstance(sub[0], list): 
        yield from sub # in python 2.x use for i in sub: yeild i 
       else:   
        yield sub 
    ....:     

In [26]: list(partial_flattening(lst)) 
Out[36]: [[40, 20, 45, 40], [20, 10, 10, 30, 45, 20], [30, 20, 20, 30]] 

如果有可能的組合然後根據您想要對數字進行的操作，您可以自定義該功能。

或者作爲一個更簡潔的方法，您可以列表解析中使用yield：

In [29]: list([(yield from i) if isinstance(i[0], list) else (yield i) for i in lst]) 
Out[29]: [[40, 20, 45, 40], [20, 10, 10, 30, 45, 20], [30, 20, 20, 30]]