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


112

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

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

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

各タプルの2番目の要素のみをリストに抽出したいと思います。

seconds = [1, 3, 5]

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

回答:



34

これも機能します:

zip(*elements)[1]

(私はこれを主に投稿しています、私が理解したことを自分に証明するためにzip...)

実際に見てください:

>>> help(zip)

builtinモジュールの組み込み関数zipに関するヘルプ:

zip(...)

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))してタプルのリストを返します。


2
**dictキーワード引数の作成に使用:def test(foo=3, bar=3): return foo*bar次にd = {'bar': 9, 'foo'=12}; print test(**d)
Wayne Werner、

@ウェインヴェルナー:うん。このようなものは、すべて単に受動的な知識(私は多くの場合、それを使用しないでください)だった-しかし、あなたは知っているので、それの良い今して思い出したことがどこ/何を探して...
ダレントーマス

1
実話-私は何で私が頻繁に十分使用することを見つける(パイソン、Vimは)、私はきちんと/クールの必要性リマインダに傾向がある私はそれらを使用していないので、私は忘れてしまったことを特徴とすることをしばしば。
Wayne Werner

* list構文はかなり便利です。これが公式のPythonドキュメントに記載されているアイデアはありますか?
user1748155 2013年

:私はチュートリアルでそれを見つけたdocs.python.org/2/tutorial/...
ダレントーマス

30

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

別の方法があります。マップ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

リストを反復処理する必要がある場合は、aを使用するforと問題ありません。


2
小さな追加:python-3.xでは、ベンチマークは、マップがほんの数ミリ秒しかかからないことを示します。それはイテレータを返すからです。method2 = 'list(map(itemgetter(1)、elements))'は、古い動作をレンダリングします。
Maik Beckmann、

12

これを見つけたのは、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つのタプルのペアがあり、dictに変換して値を取得する場合、2倍以上の速さになります。


これはおそらく明白ですがdict(elements).values()、リスト内包表記やマップとは対照的に、1要素のdictになることに言及します。これはまさに私が欲しかったものです(私はユニークなtouplesに興味がありました)(+1と投稿への大きな感謝)。
Greg0ry 2016

6

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

また、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()ください。したがって、明示的な変換です。



1

使用するislicechain.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]
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.