Pythonで別のリストの値を使ってリストをマスクする方法

あるリストの各要素が、別のリスト（検索対象リスト）に含まれているかどうかに応じて、元のリストを0と1だけで構成される「マスク」に変換したいケースはよくあります。このような処理には、Pythonのリスト内包表記を使うのが最もシンプルで読みやすい方法です。

サンプルコード

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

my_list = [5, 6, 1, 9, 11, 0, 4]

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

search_list = [2, 10, 6, 3, 9]

result = [1 if element in search_list else 0 for element in my_list]

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

出力結果

The list is :
[5, 6, 1, 9, 11, 0, 4]
The result is :
[0, 1, 0, 1, 0, 0, 0]

コードの解説

• まず、マスク対象となるリストmy_listを定義し、コンソールに表示します。

• 次に、検索条件となる別のリストsearch_listを定義します。

• リスト内包表記を使ってmy_listの各要素を順番に取り出し、その要素がsearch_listに存在すれば1、存在しなければ0を生成します。

• 生成されたマスク（0と1のリスト）は変数resultに代入されます。

• 最後に、結果をコンソールに出力して確認します。

補足：パフォーマンスを上げるテクニック

search_listが大きくなると、リストに対するin演算子の判定は線形時間（O(n)）を要するため、処理速度が低下する可能性があります。あらかじめset（集合型）に変換しておけば、要素の存在確認を平均O(1)で行えるようになり、大幅な高速化が期待できます。

search_set = set(search_list)
result = [1 if element in search_set else 0 for element in my_list]