Jak wyodrębnić n-te elementy z listy krotek w python?

Question

Próbuję uzyskać n-te elementy z listy krotek.

mam coś takiego:

elements = [(1,1,1),(2,3,7),(3,5,10)] 

życzę, aby wyodrębnić tylko drugi elementów każdej krotki do listy:

seconds = [1, 3, 5] 

Wiem, że można to zrobić z for pętli ale Chciałem wiedzieć, czy jest inny sposób, ponieważ mam tysiące krotek.

Answer 1

[x[1] for x in elements] 

Answer 2

Wiem, że można to zrobić z za, ale chciałem wiedzieć, czy istnieje inny sposób

Jest też inny sposób. Można również zrobić to z map i itemgetter:

>>> from operator import itemgetter 
>>> map(itemgetter(1), elements) 

ten nadal wykonuje pętlę wewnętrznie chociaż i to jest nieco wolniejszy niż listowego:

setup = 'elements = [(1,1,1) for _ in range(100000)];from operator import itemgetter' 
method1 = '[x[1] for x in elements]' 
method2 = 'map(itemgetter(1), elements)' 

import timeit 
t = timeit.Timer(method1, setup) 
print('Method 1: ' + str(t.timeit(100))) 
t = timeit.Timer(method2, setup) 
print('Method 2: ' + str(t.timeit(100))) 

Wyniki:

 
Method 1: 1.25699996948 
Method 2: 1.46600008011 

Jeśli potrzebujesz do iteracji na liście, to korzystanie z for jest w porządku.

Answer 3

Działa to również:

zip(*elements)[1] 

(ja głównie delegowania to, aby udowodnić sobie, że mam groked zip ...)

Zobacz go w akcji:

>>> help(zip) 

Pomoc dotycząca wbudowanego zipu funkcji w module wbudowany:

błyskawiczny (...)

błyskawiczny (SEQ1 [, SEQ2 [...]]) -> [(SEQ1 [0] SEQ2 [0], ...), (...) ]

Zwraca listę krotek, gdzie każda krotka zawiera element i-th z każdej z sekwencji argumentów. Zwrócona lista jest skrócona o w długości do najkrótszej sekwencji argumentów.

>>> elements = [(1,1,1),(2,3,7),(3,5,10)] 
>>> zip(*elements) 
[(1, 2, 3), (1, 3, 5), (1, 7, 10)] 
>>> zip(*elements)[1] 
(1, 3, 5) 
>>> 

Neat rzecz dowiedziałem dzisiaj: Użyj *list w argumenty, aby utworzyć listę parametrów dla funkcji ...

Answer 4

Znaleziony to jak szukałem w jaki sposób jest najszybszy wyciągnąć drugi element z 2-krotnej listy.Nie to, co chciałem, ale biegł sam test jak pokazano z 3rd metody oraz przetestować metodę ZIP

setup = 'elements = [(1,1) for _ in range(100000)];from operator import itemgetter' 
method1 = '[x[1] for x in elements]' 
method2 = 'map(itemgetter(1), elements)' 
method3 = 'dict(elements).values()' 
method4 = 'zip(*elements)[1]' 

import timeit 
t = timeit.Timer(method1, setup) 
print('Method 1: ' + str(t.timeit(100))) 
t = timeit.Timer(method2, setup) 
print('Method 2: ' + str(t.timeit(100))) 
t = timeit.Timer(method3, setup) 
print('Method 3: ' + str(t.timeit(100))) 
t = timeit.Timer(method4, setup) 
print('Method 4: ' + str(t.timeit(100))) 

Method 1: 0.618785858154 
Method 2: 0.711684942245 
Method 3: 0.298138141632 
Method 4: 1.32586884499 

więc ponad dwa razy szybciej, jeśli masz parę 2 krotny po prostu przekonwertować do dict i przyjmować wartości.

Answer 5

map (lambda x:(x[1]),elements) 

Answer 6

taktowanie dla Python 3,6 ekstrakcji drugi element z listy 2-krotnego.

Dodana została również metoda tablicowa, która jest łatwiejsza do odczytania (ale prawdopodobnie prostsza niż zrozumienie listy).

from operator import itemgetter 
elements = [(1,1) for _ in range(100000)] 

%timeit second = [x[1] for x in elements] 
%timeit second = list(map(itemgetter(1), elements)) 
%timeit second = dict(elements).values() 
%timeit second = list(zip(*elements))[1] 
%timeit second = np.array(elements)[:,1] 

i czasy:

list comprehension: 4.73 ms ± 206 µs per loop 
list(map):   5.3 ms ± 167 µs per loop 
dict:    2.25 ms ± 103 µs per loop 
list(zip)   5.2 ms ± 252 µs per loop 
numpy array:  28.7 ms ± 1.88 ms per loop 

Zauważ, że map() i zip() nie zwróci listę już, stąd wyraźny konwersję.