【Python】リスト内包表記とall()関数で範囲内の要素を含む行をフィルタリングする方法

はじめに

Pythonでは、ネストされたリスト（リストのリスト）の中から、指定した範囲のすべての要素を含む行だけを抽出したいケースがあります。このような場合、リスト内包表記と組み込み関数all()を組み合わせることで、簡潔かつ効率的にフィルタリングを実現できます。

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

サンプルコード

my_list = [[3, 2, 4, 5, 10], [32, 12, 4, 51, 10], [12, 53, 11], [2, 3, 31, 5, 8, 7]]

print("元のリスト：")
print(my_list)

i, j = 2, 5

my_result = [index for index in my_list if all(element in index for element in range(i, j + 1))]

print("フィルタリング結果：")
print(my_result)

実行結果

元のリスト：
[[3, 2, 4, 5, 10], [32, 12, 4, 51, 10], [12, 53, 11], [2, 3, 31, 5, 8, 7]]
フィルタリング結果：
[[3, 2, 4, 5, 10]]

コードの解説

• まず、リストのリスト（ネストされたリスト）を定義し、コンソールに表示します。
• 次に、範囲の境界となる整数「i」と「j」の値を定義します。
• リスト内包表記を使って元のリストを反復処理し、all()関数によって各サブリストのすべての要素が、先ほど定義した2つの整数で指定された範囲に含まれているかどうかを確認します。
• 条件を満たすサブリストのみが新しいリストとして生成されます。
• その結果は変数 my_result に代入され、最後にコンソールへ出力されます。

ポイント解説

all()関数：引数として渡されたイテラブルのすべての要素が真（True）である場合に True を返します。ここでは、「範囲内の各要素がサブリストに存在するかどうか」を判定するために使用しています。

range(i, j + 1)：i から j までの連続した整数列を生成します。range() の終了値は範囲に含まれないため、j 自体もチェック対象に含めたい場合は「j + 1」を指定する必要があります。

この手法を使えば、for文を何重にも書くことなく、読みやすく保守性の高いコードで条件に合致する行の抽出が可能になります。