Jak przekonwertować listę krotek na wiele list?


110

Załóżmy, że mam listę krotek i chcę przekonwertować je na wiele list.

Na przykład lista krotek to

[(1,2),(3,4),(5,6),]

Czy w Pythonie jest jakaś funkcja wbudowana, która konwertuje ją na:

[1,3,5],[2,4,6]

To może być prosty program. Ale jestem po prostu ciekawy istnienia takiej wbudowanej funkcji w Pythonie.

Odpowiedzi:


164

Wbudowana funkcja zip()zrobi prawie to, co chcesz:

>>> zip(*[(1, 2), (3, 4), (5, 6)])
[(1, 3, 5), (2, 4, 6)]

Jedyną różnicą jest to, że zamiast list otrzymujesz krotki. Możesz przekonwertować je na listy za pomocą

map(list, zip(*[(1, 2), (3, 4), (5, 6)]))

40

Z dokumentacji Pythona :

zip () w połączeniu z operatorem * może służyć do rozpakowania listy:

Konkretny przykład:

>>> zip((1,3,5),(2,4,6))
[(1, 2), (3, 4), (5, 6)]
>>> zip(*[(1, 2), (3, 4), (5, 6)])
[(1, 3, 5), (2, 4, 6)]

Lub, jeśli naprawdę chcesz listy:

>>> map(list, zip(*[(1, 2), (3, 4), (5, 6)]))
[[1, 3, 5], [2, 4, 6]]


4

franklsf95 stawia na wydajność w swojej odpowiedzi i wybiera list.append(), ale nie są one optymalne.

Dodając listy składane, otrzymałem:

def t1(zs):
    xs, ys = zip(*zs)
    return xs, ys

def t2(zs):
    xs, ys = [], []
    for x, y in zs:
        xs.append(x)
        ys.append(y)
    return xs, ys

def t3(zs):
    xs, ys = [x for x, y in zs], [y for x, y in zs]
    return xs, ys

if __name__ == '__main__':
    from timeit import timeit
    setup_string='''\
N = 2000000
xs = list(range(1, N))
ys = list(range(N+1, N*2))
zs = list(zip(xs, ys))
from __main__ import t1, t2, t3
'''
    print(f'zip:\t\t{timeit('t1(zs)', setup=setup_string, number=1000)}')
    print(f'append:\t\t{timeit('t2(zs)', setup=setup_string, number=1000)}')
    print(f'list comp:\t{timeit('t3(zs)', setup=setup_string, number=1000)}')

Dało to wynik:

zip:            122.11585397789766
append:         356.44876132614047
list comp:      144.637765085659

Więc jeśli zależy ci na wydajności, prawdopodobnie powinieneś użyć, zip()chociaż listy składane nie są zbyt daleko w tyle. W porównaniu z nim wydajność appendjest właściwie dość słaba.


1

Pomimo *ziptego, że jest bardziej Pythonic, poniższy kod ma znacznie lepszą wydajność:

xs, ys = [], []
for x, y in zs:
    xs.append(x)
    ys.append(y)

Ponadto, gdy oryginalna lista zsjest pusta, *zipzostanie podniesiona, ale ten kod może poprawnie obsłużyć.

Właśnie przeprowadziłem szybki eksperyment i oto wynik:

Using *zip:     1.54701614s
Using append:   0.52687597s

Uruchomienie go wiele razy appendjest 3x - 4x szybsze niż zip! Skrypt testowy jest tutaj:

#!/usr/bin/env python3
import time

N = 2000000
xs = list(range(1, N))
ys = list(range(N+1, N*2))
zs = list(zip(xs, ys))

t1 = time.time()

xs_, ys_ = zip(*zs)
print(len(xs_), len(ys_))

t2 = time.time()

xs_, ys_ = [], []
for x, y in zs:
    xs_.append(x)
    ys_.append(y)
print(len(xs_), len(ys_))

t3 = time.time()

print('Using *zip:\t{:.8f}s'.format(t2 - t1))
print('Using append:\t{:.8f}s'.format(t3 - t2))

Moja wersja Pythona:

Python 3.6.3 (default, Oct 24 2017, 12:18:40)
[GCC 4.2.1 Compatible Apple LLVM 8.1.0 (clang-802.0.42)] on darwin
Type "help", "copyright", "credits" or "license" for more information.

1

Oprócz odpowiedzi Claudiu możesz użyć:

>>>a, b = map(list, zip(*[(1, 2), (3, 4), (5, 6)]))
>>>a
[1,3,5]
>>>b
[2,4,6]

Edytowane według @Peyman mohseni kiasari


1
Nie! da ci, (1, 3, 5)a (2, 4, 6)nie listy. powinieneś użyćmap(list, zip(*[(1, 2), (3, 4), (5, 6)]))
Peyman

0

Dodając do odpowiedzi Claudiu i Claudiu, a ponieważ mapa musi zostać zaimportowana z narzędzi itertools w Pythonie 3, używasz również funkcji list, takich jak:

[[*x] for x in zip(*[(1,2),(3,4),(5,6)])]
>>> [[1, 3, 5], [2, 4, 6]]
Korzystając z naszej strony potwierdzasz, że przeczytałeś(-aś) i rozumiesz nasze zasady używania plików cookie i zasady ochrony prywatności.
Licensed under cc by-sa 3.0 with attribution required.