Teilliste Abflachung in Python

Question

Ich habe ein kleines Problem, bei dem mir einiges an Python-Erfahrung fehlt. Unter der Annahme einer Liste:

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

ich etwas, das ich ‚teilweise Abflachung‘ nennen wollen, weil die erwartete Ausgabe ist:

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

Nachricht vor der lentgh von list war 2, während die erwartete Ausgabe Länge von 3. Das Problem ist, dass ich nicht im Voraus weiß, welches Element von list eine Schicht tiefer verschachtelt ist.

This answer und viele andere half nicht viel, weil die Abflachung zu weit geht. Im Anschluss an die Frage der Ausgabe lautet:

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

(die fehlenden Klammern für die ersten 4 Elemente aus der Liste merken

Answer 1

Sie können eine rekursive Funktion erstellen, testen, ob all die inneren Elemente sind Listen Wenn ja.. , den Algorithmus rekursiv anzuwenden, da sonst die Liste ergeben sich

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]] 

Or yield from (Python 3.3 und höher).

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 

oder ohne Generatoren:

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

Wenn Ihre Listen können nur Zahlen enthält oder Teillisten (nicht von ihnen beide) Sie verwenden können Generatorfunktion folgende:

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]] 

Wenn Kombination sein kann von beiden Arten dann basierend auf, was Sie mit den Zahlen tun möchten, können Sie die Funktion anpassen.

Oder als prägnanter Ansatz, den Sie yield innerhalb Liste Verständnis verwenden können:

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]]