【Python入門】N番目までのサブリストから要素を取得する3つの方法

このチュートリアルでは、ネストされたリスト（サブリストのリスト）から、N番目までのサブリストの要素を取得するプログラムの作成方法を解説します。

例として、次のような5つのサブリストを持つリストを考えてみましょう。

[['Python', 'Java'], ['C', 'Pascal'], ['Javascript', 'PHP'], ['C#', 'C++'], ['React', 'Angular']]

ここでの目標は、最初の3つのサブリストから、それぞれの先頭の要素を取り出すことです。実現方法はいくつかあるため、代表的なアプローチを順番に見ていきましょう。

方法1：forループを使う

多くのプログラマーが真っ先に思いつくのが、forループを使ったもっとも汎用的な方法です。以下のコードを見てください。

コード例

# リストと N を初期化
random_list = [['Python', 'Java'], ['C', 'Pascal'], ['Javascript', 'PHP'], ['C#', 'C++'], ['React', 'Angular']]
N = 3

# 最終的な要素を格納する空のリスト
desired_elements = []

# 先頭から3つ目までのサブリストを反復処理
for i in range(N):
    # 各サブリストの先頭要素を格納
    desired_elements.append(random_list[i][0])

# 結果を出力
print(desired_elements)

実行結果

上記のコードを実行すると、次の出力が得られます。

['Python', 'C', 'Javascript']

インデックス i を使って各サブリストにアクセスし、さらに [0] で先頭の要素を取得しています。シンプルで理解しやすいのが特徴です。

方法2：リスト内包表記を使う

forループの代わりにリスト内包表記を使うと、より簡潔に書けます。同じ処理を1行で実現できます。

コード例

# リストと N を初期化
random_list = [['Python', 'Java'], ['C', 'Pascal'], ['Javascript', 'PHP'], ['C#', 'C++'], ['React', 'Angular']]
N = 3

# スライスで最初の N 個のサブリストを取得し、各サブリストの先頭要素を抜き出す
desired_elements = [sublist[0] for sublist in random_list[:N]]

# 結果を出力
print(desired_elements)

実行結果

['Python', 'C', 'Javascript']

random_list[:N] のスライスで対象となるサブリストを絞り込み、その各要素から先頭の値を取り出しています。Pythonらしい読みやすい書き方です。

方法3：組み込みモジュールを活用する

Pythonには便利な組み込みモジュールや関数が数多く用意されています。ここでは、map・itemgetter・islice を組み合わせて、同じ結果を効率的に得る方法を紹介します。

• operator.itemgetter(0)：オブジェクトの0番目の要素を取得する関数を返します
• itertools.islice：イテラブルから指定した範囲だけを効率よく切り出します
• map：各サブリストに対して itemgetter を適用します

コード例

# 必要なメソッドをインポート
import operator   # itemgetter 用
import itertools  # islice 用

# リストと N を初期化
random_list = [['Python', 'Java'], ['C', 'Pascal'], ['Javascript', 'PHP'], ['C#', 'C++'], ['React', 'Angular']]
N = 3

# 最初の N 個のサブリストから先頭要素を取得
desired_elements = list(map(operator.itemgetter(0), itertools.islice(random_list, N)))

# 結果を出力
print(desired_elements)

実行結果

['Python', 'C', 'Javascript']

この方法は、大規模なデータを扱う際にイテレータベースの処理ができるため、メモリ効率の面で有利です。

まとめ

今回は、ネストされたリストからN番目までのサブリストの要素を取得する3つの方法を紹介しました。

• forループ：初心者にもわかりやすく、柔軟な処理が可能
• リスト内包表記：簡潔でPythonらしい書き方
• map + itemgetter + islice：イテレータを活用した効率的な方法

なお、本記事では先頭の要素を取得しましたが、インデックスを変更すれば任意の位置の要素を同様に取得できます。状況に応じて最適な方法を選んでみてください。チュートリアルについて不明な点があれば、コメント欄でお気軽にご質問ください。