Python - elegante Art und Weise Liste aller Kombinationen zu verwandeln

Question

ich eine Liste von Elementen wie habe:

x = ['A, B', '2', '3', 'Jan, Feb'] 
# ^    ^
# comma separated single string 

Ich mag diese Liste in eine Liste von Listen mit allen Kombinationen der einzelnen Elemente zu transformieren das heißt durch Kommas getrennt, so dass es wie folgt aussieht:

[['A', '2', '3', 'Jan'], ['A', '2', '3', 'Feb'] 
['B', '2', '3', 'Jan'], ['B', '2', '3', 'Feb']] 

Was ist die elegante Möglichkeit, dies in integrierten Funktionen/Bibliotheken mit Python zu erreichen?

Answer 1

Sie können zunächst die Elemente der Liste verteilen basierend auf Komma , und dann itertools.product verwenden, um die gewünschte Liste zu erhalten, wie:

from itertools import product 
x = ['A, B', '2', '3', 'Jan, Feb'] 

new_list = list(product(*[i.split(', ') for i in x])) 

wo Inhalte new_list wird:

[('A', '2', '3', 'Jan'), 
('A', '2', '3', 'Feb'), 
('B', '2', '3', 'Jan'), 
('B', '2', '3', 'Feb')] 

---

Basierend auf Dan Ds Kommentar, unten ist der timeit Vergleich der Zwischenliste V/s Zwischengenerator Vergleich auf Python 2. Zwischenliste Ausdruck ist etwas schneller:

mquadri$ python -m timeit "from itertools import product; list(product(*(i.split(', ') for i in ['A, B', '2', '3', 'Jan, Feb'])))" 
100000 loops, best of 3: 4.83 usec per loop 

mquadri$ python -m timeit "from itertools import product; list(product(*[i.split(', ') for i in ['A, B', '2', '3', 'Jan, Feb']]))" 
100000 loops, best of 3: 3.78 usec per loop