【Python】リスト内のタプルからn番目の要素を取得する方法

Pythonのリストには、タプルを要素として格納することができます。本記事では、リスト内の各タプルからn番目の要素を取り出す方法を、具体的なサンプルコードと実行結果とともに解説します。

方法1:インデックスを使う

リスト内包表記を使えば、forループでリストの各タプルを走査しながら、指定したインデックスの要素に直接アクセスできます。以下の例では、各タプルのインデックス1(2番目)の要素を取り出し、新しいリストに結果を格納しています。

サンプルコード

Alist = [('Mon','3 pm',10),('Tue','12pm',8),('Wed','9 am',8),('Thu','6 am',5)]

# 元のリストを表示
print("Given list: ", Alist)

# インデックス1の要素を取得
res = [x[1] for x in Alist]

print("各タプルの2番目の要素:\n", res)

実行結果

上記のコードを実行すると、次のような結果が得られます。

Given list: [('Mon', '3 pm', 10), ('Tue', '12pm', 8), ('Wed', '9 am', 8), ('Thu', '6 am', 5)]
各タプルの2番目の要素:
['3 pm', '12pm', '9 am', '6 am']

方法2:itemgetterを使う

operatorモジュールのitemgetter関数は、指定したインデックス位置の要素を取得するための呼び出し可能オブジェクトを返します。map関数と組み合わせることで、リスト内のすべてのタプルに対して同じ操作を一括で適用でき、最後にlist関数で結果をリスト化します。以下の例では、各タプルのインデックス2(3番目)の要素を取得しています。

サンプルコード

from operator import itemgetter

Alist = [('Mon','3 pm',10),('Tue','12pm',8),('Wed','9 am',8),('Thu','6 am',5)]

# 元のリストを表示
print("Given list: ", Alist)

# itemgetterでインデックス2の要素を取得
res = list(map(itemgetter(2), Alist))

print("各タプルの3番目の要素:\n", res)

実行結果

上記のコードを実行すると、次のような結果が得られます。

Given list: [('Mon', '3 pm', 10), ('Tue', '12pm', 8), ('Wed', '9 am', 8), ('Thu', '6 am', 5)]
各タプルの3番目の要素:
[10, 8, 8, 5]

まとめ

リスト内のタプルからn番目の要素を取得するには、リスト内包表記とoperator.itemgetterの2つの方法が有効です。コードの可読性を重視するならリスト内包表記、大量データを扱う際の処理効率や関数型スタイルを重視するならitemgetterがおすすめです。目的や規模に応じて使い分けるとよいでしょう。