Append Elemente eines willkürlich verschachtelt Liste in einer Tiefensuche-artige Weise ohne Rekursion

Question

Die Liste tiefer sein kann, oder flacher, aber ich habe eine Liste der Tiefe 2 haben, wie folgt:

a = [['a','b'],['c','d'],['e','f']] 

I wollen eine Funktion schreiben, f(a), so dass sie die folgende neue Liste zurückkehren würde:

['acef', 'adef', 'bcef', 'bdef'] 

ich imitiert Tiefensuche, wo die Listen Knoten sind im Wesentlichen. Ich möchte, dass die Funktion mit depth = n arbeitet, wobei n eine beliebige ganze Zahl ist. Was ist der pythischste Weg, dies zu erreichen?

Mein rekursiven Code lautet wie folgt:

def f(elems): 
    curr, *rest = elems 

    if not rest: 
     return ''.join(curr) 

    ret = [''.join(x + f(rest)) for x in curr] 
    return ret 

Wie würde ich mich über diese iterativ zu lösen?

Answer 1

können Sie itertools.product verwenden:

import itertools 

def f(elems): 
    *branches, leaves = elems 
    for path in itertools.product(*branches): 
     yield ''.join(itertools.chain(path, leaves)) 

a = [['a','b'],['c','d'],['e','f']] 
print(list(f(a))) 

Das gibt:

['acef', 'adef', 'bcef', 'bdef']