Pythonでネストされたリスト（サブリスト）の要素のインデックスを取得する方法

はじめに

このチュートリアルでは、ネストされたリスト（リストの中にリストが入った構造）から、特定の要素が含まれるサブリストのインデックスを見つけるプログラムを作成します。まずは具体例を見て、動作をしっかり理解しましょう。

入力

nested_list = [[1, 2, 3], [4, 5], [6, 7, 8, 9]]

出力

7 のインデックス： 2
5 のインデックス： 1
3 のインデックス： 0

解決の手順

この問題は、シンプルかつ最も一般的な方法で解くことができます。以下の手順に従って進めてください。

• ネストされたリストを初期化する。
• インデックスを使って外側のリストを反復処理する。
• 各サブリストを反復処理し、目的の要素と一致するかどうかを確認する。
• 要素が見つかったら、そのサブリストのインデックスを出力してループを抜ける。

コード例

# リストの初期化
nested_list = [[1, 2, 3], [4, 5], [6, 7, 8, 9]]

# インデックスを見つける関数
def find_index(element):
    # 要素が見つかったかどうかを追跡するフラグ
    is_found = False
    # 外側のリストを反復処理
    for i in range(len(nested_list)):
        # サブリストを反復処理
        for j in range(len(nested_list[i])):
            # 要素と一致するか確認
            if nested_list[i][j] == element:
                # 要素を含むサブリストのインデックスを出力
                print(f'{element} のインデックス： {i}')
                # フラグを True に変更
                is_found = True
                # 内側のループを抜ける
                break
        # 見つかった場合は外側のループも抜ける
        if is_found:
            break
    # 要素が見つからなかった場合
    if not is_found:
        print("要素はリストに存在しません")

find_index(7)
find_index(5)
find_index(3)

実行結果

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

7 のインデックス： 2
5 のインデックス： 1
3 のインデックス： 0

応用：サブリスト内の位置も一緒に取得する

サブリストのインデックスに加えて、そのサブリスト内での要素の位置も知りたい場合は、enumerate を使うとより簡潔に書けます。

nested_list = [[1, 2, 3], [4, 5], [6, 7, 8, 9]]

def find_position(element):
    for i, sub_list in enumerate(nested_list):
        if element in sub_list:
            return i, sub_list.index(element)
    return None

result = find_position(8)
print(f"サブリストのインデックス： {result[0]}")
print(f"サブリスト内の位置： {result[1]}")

実行結果

サブリストのインデックス： 2
サブリスト内の位置： 1

まとめ

このチュートリアルでは、ネストされたリストから特定の要素を含むサブリストのインデックスを見つける方法を学びました。二重の for ループを使った基本的なアプローチと、enumerate を活用したより Pythonic な書き方の両方を紹介しました。状況に応じて使い分けてみてください。チュートリアルについて質問がある場合は、コメント欄でお気軽にお尋ねください。