Pythonでリスト内の要素を検索する方法：初心者向け完全ガイド

Pythonのリストから特定の要素を見つけるにはどうすればよいのでしょうか？これはPythonプログラマーなら誰もが一度は直面する疑問です。幸い、Pythonにはin演算子や線形探索など、リスト内の要素を検索するためのさまざまな手法が用意されています。

本記事では、リストから要素を検索する4つの方法を紹介し、それぞれの動作を実例とともにわかりやすく解説します。

1. 「in」演算子を使った検索

Pythonにはinという特別な演算子があります。この演算子は、ある値がリストに含まれているかどうかをチェックするもので、「メンバーシップ演算子」と呼ばれることもあります。

例として、靴店を経営していて、特定の靴が在庫にあるかどうかを確認したいケースを考えてみましょう。in演算子を使えば、簡単に実現できます。

まず、靴のリストを定義します。

shoes = ["Adidas ZX Flux", "Adidas Ultraboost", "Adidas Gazelle", "Adidas Runfalcon"]

次に、ユーザーに検索したい靴の名前を入力してもらいます。

to_find = input("What shoe are you looking for? ")

input()関数で受け取った靴の名前をもとに、if文でその靴がリストに存在するかどうかを判定します。

if to_find in shoes:
    print("{} shoes are in stock.".format(to_find))
else:
    print("{} shoes are not in stock.".format(to_find))

ユーザーが入力した靴がリスト内に見つかればif文が実行され、見つからなければelse文が実行されます。実際にコードを実行してみましょう。

What shoe are you looking for? Adidas Runfalcon
Adidas Runfalcon shoes are in stock.

コードは「Adidas Runfalcon」が在庫にあることを正しく判定し、if文が実行されています。

2. 線形探索（Linear Search）を使った検索

線形探索は、リストから要素を見つけるためのシンプルな検索アルゴリズムです。リストの先頭から順に各要素と目的の値を比較していき、値が見つかればその位置（インデックス）を返します。

まず、線形探索を行う関数を作成しましょう。

def linear_search(array, to_find):
    for i in range(0, len(array)):
        if array[i] == to_find:
            return i
    return -1

このコードでは、forループを使ってarrayリスト内のすべての要素を反復処理します。該当する要素が見つかった場合は、そのインデックス位置を返します。リスト全体を検索しても見つからなかった場合は、-1を返します。

続いて、メインプログラム側でこの関数を呼び出します。

shoes = ["Adidas ZX Flux", "Adidas Ultraboost", "Adidas Gazelle", "Adidas Runfalcon"]
to_find = input("What shoe are you looking for? ")

found = linear_search(shoes, to_find)

このコードはユーザーに探している靴を尋ね、linear_search関数を呼び出して、その靴がリストに存在するかどうかを確認します。

最後に、if文で検索結果をユーザーに通知します。

if found != -1:
    print("{} shoes are in stock.".format(to_find))
else:
    print("{} shoes are not in stock.".format(to_find))

linear_search関数が-1を返した場合、それは要素がリスト内に存在しなかったことを意味します。戻り値が-1以外であればif文が、-1であればelse文が実行される仕組みです。

コードを実行してみます。

What shoe are you looking for? Adidas Samba
Adidas Samba shoes are not in stock.

リスト内に「Adidas Samba」は存在しないため、在庫なしという結果が表示されました。

3. index()メソッドを使った検索

組み込みメソッドindex()を使うと、リスト内の要素のインデックス位置を直接取得できます。ここでは、index()を使って靴の位置を特定するプログラムを書いてみましょう。

まず、靴のリストを定義し、ユーザーに検索対象の靴を入力してもらいます。

shoes = ["Adidas ZX Flux", "Adidas Ultraboost", "Adidas Gazelle", "Adidas Runfalcon"]
to_find = input("What shoe are you looking for? ")

次に、index()メソッドでその靴のリスト内での位置を取得します。

try:
    found = shoes.index(to_find)
    print("{} shoes are in stock. They are at index position {} in the list of shoes.".format(to_find, found))
except:
    print("{} shoes are not in stock.".format(to_find))

index()メソッドは、要素がリスト内に見つからない場合にValueErrorを発生させます。そのため、ここでは「try…except」ブロックの中で使用しています。

靴が見つからなければValueErrorが発生し、exceptブロックが実行されます。逆に見つかれば、tryブロックが正常に完了します。

コードを実行すると、次のような結果になります。

What shoe are you looking for? Adidas Gazelle
Adidas Gazelle shoes are in stock. They are at index position 2 in the list of shoes.

コードは「Adidas Gazelle」が在庫にあることを正しく判定するだけでなく、リスト内での保存位置（インデックス2）まで教えてくれています。

4. リスト内包表記を使った検索

リスト内包表記を使うと、特定の条件を満たす複数の要素をまとめて抽出できます。

ここでは、靴のリストからNikeブランドのものだけを取り出す例を見てみましょう。まず、靴のリストを定義します。

shoes = ["Nike PG 4", "Adidas Ultraboost", "Adidas Gazelle", "Nike Air Max Verona"]

次に、「Nike」という文字列を含むすべての靴を取得するリスト内包表記を書きます。リスト内包表記は、forループに似た構文を使用します。

nike_shoes = [shoe for shoe in shoes if "Nike" in shoe]
print(nike_shoes)

この式はshoesリスト内のすべての靴を反復処理し、各靴の名前に「Nike」が含まれているかどうかをチェックします。条件を満たす靴だけがnike_shoesリストに追加されます。

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

['Nike PG 4', 'Nike Air Max Verona']

Nikeブランドの靴だけをまとめたリストが返されました。

まとめ

Pythonでリストから要素を検索する方法は多数ありますが、代表的なのは以下の4つです。

• inメンバーシップ演算子を使う方法
• 線形探索（Linear Search）を使う方法
• index()メソッドを使う方法
• リスト内包表記を使う方法

それぞれの特徴を理解して使い分ければ、状況に応じた最適な検索が可能になります。これであなたも、プロのコーダーのようにPythonリスト内の要素を自在に検索できるようになったはずです！