Python – 指定したリストの要素をすべて含む行を抽出する方法

ネストされたリスト（リストのリスト）の中から、特定のリストの要素をすべて含む行だけを取り出したい場面はよくあります。この記事では、フラグ変数、シンプルな反復処理、そして「append」メソッドを組み合わせた基本的な実装方法を解説します。

サンプルコード

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

my_list = [[8, 6, 3, 2], [1, 6], [2, 1,7], [8, 1, 2]]

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

sub_list = [1, 2]

result = []
for row in my_list:

   flag = True

   for element in sub_list:
      if element not in row:
         flag = False
   if flag:
      result.append(row)

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

出力結果

The list is :
[[8, 6, 3, 2], [1, 6], [2, 1, 7], [8, 1, 2]]
The resultant list is :
[[2, 1, 7], [8, 1, 2]]

処理の流れと解説

• まず、リストのリスト（ネストされたリスト）を定義し、コンソールに表示します。

• 次に、検索対象となる整数値を含む別のリスト（sub_list）を定義します。

• 結果を格納するための空のリストを用意します。

• 外側のループでリストの各行を順番に取り出し、フラグ変数を「True」で初期化します。

• 内側のループでsub_listの各要素を確認し、その要素が現在の行に存在しない場合は、フラグを「False」に設定します。

• ループ終了後、フラグの値によって処理を分岐させます。

• フラグが「True」のまま（＝すべての要素が含まれている）であれば、その行を結果リストに追加（append）します。

• 最後に、結果として得られたリストをコンソールに出力します。

より簡潔な書き方：all() 関数を活用する

上記のフラグ変数を使う方法は理解しやすい反面、コードがやや長くなります。Pythonでは組み込み関数 all() を使うことで、同じ処理を1行で簡潔に記述できます。

result = [row for row in my_list if all(element in row for element in sub_list)]

print(result)  # [[2, 1, 7], [8, 1, 2]]

all() は引数内のすべての条件が真である場合に「True」を返すため、フラグ変数を手動で管理する必要がなくなり、可読性も向上します。状況に応じて使い分けるとよいでしょう。