セットに含まれる要素だけをリストから抽出するPythonプログラム

Pythonでは、セット（set）に含まれる要素だけをリストから抽出したい場面がよくあります。そのような場合、シンプルな for ループと条件分岐（if文）を組み合わせるだけで簡単に実現できます。

サンプルコード

以下に具体的な実装例を示します。

my_list = [5, 7, 2, 7, 2, 4, 9, 8, 8]

print("The list is :")
print(my_list)

search_set = {6, 2, 8}

my_result = []
for element in my_list:

   if element in search_set:
      my_result.append(element)

print("The result is :")
print(my_result)

実行結果

The list is :
[5, 7, 2, 7, 2, 4, 9, 8, 8]
The result is :
[2, 2, 8, 8]

コードの解説

• まず、対象となるリストを定義し、コンソールに表示します。このリストには重複した要素が含まれています。
• 次に、検索条件となる特定の要素を持つセットを定義します。
• 抽出結果を格納するための空のリストを用意します。
• for ループでリストの各要素を順番に取り出し、その要素がセット内に存在するかどうかを in 演算子で判定します。
• 存在する場合は、あらかじめ用意した空のリストへ append メソッドで追加していきます。
• 最終的に、セットに含まれていた要素のみが抽出されたリストがコンソールに表示されます。

ポイント：セットを使うメリット

セットのメンバーシップ判定（in 演算子）は平均して O(1) の計算量で実行されるため、要素数が多い場合でも高速に動作します。もしリストに対して同じ判定を行うと O(n) かかるため、大規模なデータを扱う際はセットを活用するのが効率的です。

補足：リスト内包表記での簡潔な書き方

上記のコードは、リスト内包表記を使うともっと簡潔に書けます。

my_result = [element for element in my_list if element in search_set]

この1行で同じ結果が得られるため、状況に応じて使い分けるとよいでしょう。