Transponieren/Entpacken-Funktion (Inverse von zip)?

Question

Ich habe eine Liste von 2-Element-Tupeln und möchte sie in 2 Listen konvertieren, wobei die erste Liste das erste Element in jedem Tupel und die zweite Liste das zweite Element enthält.

Zum Beispiel:

original = [('a', 1), ('b', 2), ('c', 3), ('d', 4)]
# and I want to become...
result = (['a', 'b', 'c', 'd'], [1, 2, 3, 4])

Gibt es eine eingebaute Funktion, die das tut?

Answer 1

zip ist seine eigene Inverse! Vorausgesetzt, Sie verwenden den speziellen Operator *.

>>> zip(*[('a', 1), ('b', 2), ('c', 3), ('d', 4)])
[('a', 'b', 'c', 'd'), (1, 2, 3, 4)]

Dies geschieht durch den Aufruf von zip mit den Argumenten:

zip(('a', 1), ('b', 2), ('c', 3), ('d', 4))

außer dass die Argumente an zip direkt (nachdem es in ein Tupel umgewandelt wurde), so dass man sich keine Sorgen machen muss, dass die Anzahl der Argumente zu groß wird.

Answer 2

Sie könnten auch Folgendes tun

result = ([ a for a,b in original ], [ b for a,b in original ])

Es sollte besser skalieren. Vor allem, wenn Python es schafft, die Listenkomplexe nur bei Bedarf zu erweitern.

(Nebenbei bemerkt, ergibt dies ein 2-Tupel (Paar) von Listen und nicht eine Liste von Tupeln, wie zip tut.)

Wenn Generatoren anstelle von tatsächlichen Listen in Ordnung sind, würde dies ausreichen:

result = (( a for a,b in original ), ( b for a,b in original ))

Die Generatoren fressen sich nicht durch die Liste, bis Sie nach jedem Element fragen, aber andererseits behalten sie Verweise auf die ursprüngliche Liste.

Answer 3

Ich verwende gerne zip(*iterable) (das ist das Stück Code, das Sie suchen) in meinen Programmen wie folgt:

def unzip(iterable):
    return zip(*iterable)

Ich finde unzip besser lesbar.

Answer 4

Wenn Sie Listen haben, die nicht gleich lang sind, sollten Sie zip nicht wie in Patricks Antwort verwenden. Das funktioniert:

>>> zip(*[('a', 1), ('b', 2), ('c', 3), ('d', 4)])
[('a', 'b', 'c', 'd'), (1, 2, 3, 4)]

Bei Listen unterschiedlicher Länge schneidet zip jedoch jedes Element auf die Länge der kürzesten Liste ab:

>>> zip(*[('a', 1), ('b', 2), ('c', 3), ('d', 4), ('e', )])
[('a', 'b', 'c', 'd', 'e')]

Sie können map with no function verwenden, um leere Ergebnisse mit None zu füllen:

>>> map(None, *[('a', 1), ('b', 2), ('c', 3), ('d', 4), ('e', )])
[('a', 'b', 'c', 'd', 'e'), (1, 2, 3, 4, None)]

zip() ist jedoch geringfügig schneller.

Answer 5

>>> original = [('a', 1), ('b', 2), ('c', 3), ('d', 4)]
>>> tuple([list(tup) for tup in zip(*original)])
(['a', 'b', 'c', 'd'], [1, 2, 3, 4])

Liefert ein Tupel von Listen wie in der Frage.

list1, list2 = [list(tup) for tup in zip(*original)]

Packt die beiden Listen aus.