Pythonで各サブリストの最後の要素を取得する3つの方法

Pythonでは、リストの中にさらにリストを要素として持つことができます。このようにネストされたリストは「サブリスト」と呼ばれます。本記事では、与えられたリストから各サブリストの最後の要素だけを取り出す方法を、3つの異なるアプローチで解説します。

方法1: forループを使う

もっともシンプルなのがforループを使う方法です。リスト内の各サブリストを順番に処理し、Pythonの負のインデックス -1 を指定することで、末尾の要素を簡単に取得できます。

コード例

Alist = [['Mon', 1], ['Tue', 'Wed', "Fri"], [12, 3, 7]]
print("Given List:\n", Alist)
print("Last Items from sublists:\n")
for item in Alist:
    print(item[-1])

実行結果

Given List:
[['Mon', 1], ['Tue', 'Wed', 'Fri'], [12, 3, 7]]
Last Items from sublists:
1
Fri
7

方法2: zipと*（アンパック）を使う

* 演算子を使うと、サブリストをアンパックして個々の要素にアクセスできます。ここでは reversed 関数で各サブリストを逆順にしてから先頭の要素（＝元のリストの最後の要素）を取得し、最後に zip でまとめて結果をリスト化します。

コード例

Alist = [['Mon', 1], ['Tue', 'Wed', "Fri"], [12, 3, 7]]
print("Given List:\n", Alist)
print("\nLast Items from sublists:\n")
print(list(list(zip(*map(reversed, Alist)))[0]))

実行結果

Given List:
[['Mon', 1], ['Tue', 'Wed', 'Fri'], [12, 3, 7]]
Last Items from sublists:
[1, 'Fri', 7]

方法3: itemgetterを使う

operator モジュールの itemgetter(i) は、リスト・タプル・辞書などのイテラブルオブジェクトを受け取り、そのi番目の要素を返す呼び出し可能オブジェクトを生成します。itemgetter(-1) を指定すれば末尾の要素が取得できるため、map 関数と組み合わせることで簡潔に記述できます。

コード例

from operator import itemgetter

Alist = [['Mon', 1], ['Tue', 'Wed', "Fri"], [12, 3, 7]]
print("Given List:\n", Alist)
print("\nLast Items from sublists:\n")
print(list(map(itemgetter(-1), Alist)))

実行結果

Given List:
[['Mon', 1], ['Tue', 'Wed', 'Fri'], [12, 3, 7]]
Last Items from sublists:
[1, 'Fri', 7]

まとめ

用途や可読性の要件に応じて、これらの方法を使い分けるとよいでしょう。単純な処理ならforループ、ワンライナーで書きたい場合は itemgetter(-1) と map の組み合わせが特におすすめです。