web-dev-qa-db-ja.com

タプルのリストからn番目の要素を抽出する方法は？

タプルのリストからn番目の要素を取得しようとしています。

私のようなものがあります：

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

各Tupleの2番目の要素のみをリストに抽出したい：

seconds = [1, 3, 5]

forループで実行できることは知っていますが、数千のタプルがあるため、別の方法があるかどうかを知りたかったのです。

pythonlisttuples
94
pleasedontbelong
n = 1 # N. . .
[x[n] for x in elements]
163
luc

FORで実行できることは知っていますが、別の方法があるかどうか知りたい

別の方法があります。 map および itemgetter でもできます：

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

ただし、これは内部でループを実行しますが、リストの理解よりも少し遅くなります。

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)))

結果：

方法1：1.25699996948 
方法2：1.46600008011

リストを反復処理する必要がある場合は、forを使用するのが適切です。

28
Mark Byers

これも機能します：

Zip(*elements)[1]

（主にこれを投稿しているのは、自分がZip...を盗んだことを証明するためです）

実際にご覧ください：

>>> help(Zip)

モジュールの組み込み関数Zipのヘルプbuiltin：

郵便番号（...）

Zip（seq1 [、seq2 [...]]）-> [（seq1 [0]、seq2 [0] ...）、（...）]

タプルのリストを返します。各タプルには、各引数シーケンスのi番目の要素が含まれます。返されるリストの長さは、最短の引数シーケンスの長さに切り捨てられます。

>>> 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)
>>>

今日学んだ素敵なこと：引数に*listを使用して、関数のパラメーターリストを作成します...

注：Python3では、Zipはイテレータを返すので、代わりにlist(Zip(*elements))を使用してタプルのリストを返します。

25
Daren Thomas

これは、2タプルリストの2番目の要素を取得するのに最も速い方法を探していたときに見つけました。私が望んでいたものではありませんが、3番目の方法で示されているのと同じテストを実行し、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

したがって、辞書に変換して値を取得するための2タプルペアがある場合は、2倍以上の速さです。

11
Graeme Gellatly
map (lambda x:(x[1]),elements)
3
Thras

Python 3.6のタイミング。2タプルリストから2番目の要素を抽出します。

また、numpy配列メソッドを追加しました。これは読みやすくなっています（ただし、リストの内包表記よりも間違いなく単純です）。

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]

およびタイミング：

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

map()とZip()はリストを返さないため、明示的な変換が行われることに注意してください。

3
Oleg

islice および chain.from_iterable を使用：

>>> from itertools import chain, islice
>>> elements = [(1,1,1),(2,3,7),(3,5,10)]
>>> list(chain.from_iterable(islice(item, 1, 2) for item in elements))
[1, 3, 5]

これは、複数の要素が必要な場合に役立ちます。

>>> elements = [(0, 1, 2, 3, 4, 5), 
                (10, 11, 12, 13, 14, 15), 
                (20, 21, 22, 23, 24, 25)]
>>> list(chain.from_iterable(islice(Tuple_, 2, 5) for Tuple_ in elements))
[2, 3, 4, 12, 13, 14, 22, 23, 24]
1
Georgy