web-dev-qa-db-ja.com

Pythonのリストの各Tupleの最初の要素を取得する

SQLクエリは私にこのようなタプルのリストを与えます：

[(elt1, elt2), (elt1, elt2), (elt1, elt2), (elt1, elt2), (elt1, elt2), ...]

各タプルの最初の要素をすべて手に入れたいです。今私はこれを使います。

rows = cur.fetchall()
res_list = []
for row in rows:
    res_list += [row[0]]

しかし、それを実行するためのより良い構文があるかもしれないと思います。もっと良い方法を知っていますか？

pythonpython-2.7syntax
101
Creak

リスト内包表記 を使用してください。

res_list = [x[0] for x in rows]

以下はデモンストレーションです。

>>> rows = [(1, 2), (3, 4), (5, 6)]
>>> [x[0] for x in rows]
[1, 3, 5]
>>>

あるいは、x[0]の代わりに解凍を使用することもできます。

res_list = [x for x,_ in rows]

以下はデモンストレーションです。

>>> lst = [(1, 2), (3, 4), (5, 6)]
>>> [x for x,_ in lst]
[1, 3, 5]
>>>

どちらの方法でもほぼ同じことができるので、どちらを選択してもかまいません。

161
iCodez

これを実現するための機能的な方法は、以下を使用してリストを解凍することです。

sample = [(2, 9), (2, 9), (8, 9), (10, 9), (23, 26), (1, 9), (43, 44)]
first,snd = Zip(*sample)
print first,snd
(2, 2, 8, 10, 23, 1, 43) (9, 9, 9, 9, 26, 9, 44)
19
rbonallo

何らかの理由でリスト内包表記を使用したくない場合は、 map および operator.itemgetter を使用できます。

>>> from operator import itemgetter
>>> rows = [(1, 2), (3, 4), (5, 6)]
>>> map(itemgetter(1), rows)
[2, 4, 6]
>>>
19
Jimilian

リスト内包表記を使うことができます。

res_list = [i[0] for i in rows]

これでうまくいくはずです

11
Ruben Bermudez
res_list = [x[0] for x in rows]

c.f http://docs.python.org/3/tutorial/datastructures.html#list-comprehensions

mapのような高階関数よりも内包表記を好む理由については、 http://www.artima.com/weblogs/viewpost.jsp?thread=98196 を参照してください。

6
Aegis