Pythonで特定の数字のみを含む要素をリストから抽出する方法

リストの中から、特定の数字だけで構成されている数値だけを抽出したいケースはよくあります。そんなときに便利なのが、リスト内包表記とall() 関数を組み合わせた方法です。この記事では、その具体的な実装例と動作の仕組みをわかりやすく解説します。

サンプルコード

my_list = [3345, 2345, 1698, 2475, 1932]

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

digit_list = [2, 3, 5, 4]

my_result = [index for index in my_list if all(int(element) in digit_list for element in str(index))]

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

実行結果

The list is :
[3345, 2345, 1698, 2475, 1932]
The result is :
[3345, 2345]

コードの解説

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

• 次に、抽出条件となる数字のリスト digit_list を定義します。この例では [2, 3, 5, 4] を指定しており、これらの数字だけで構成される数値のみが抽出対象になります。

• リスト内包表記を使って my_list の各要素を順番に取り出し、str() 関数で文字列に変換します。数値を文字列化することで、1桁ずつ個別に扱えるようになります。

• 各桁に対して int() で再び整数に戻し、その数字が digit_list に含まれているかどうかを確認します。

• all() 関数により、すべての桁が digit_list に含まれる場合のみ True となり、その要素が結果のリストに追加されます。

• 最終的な結果は変数 my_result に格納され、コンソールに出力されます。

この手法のポイント

この方法の核心は、数値を一度文字列に変換することで「桁」を簡単に走査できる点にあります。また、all() はイテラブル内のすべての要素が真である場合に True を返すため、「すべての桁が条件を満たしているか」という判定を1行で簡潔に記述できます。

たとえば上記の例では、3345 と 2345 は使用できる数字（2, 3, 5, 4）だけで構成されているため抽出されましたが、1698 や 1932 には条件外の数字（1, 6, 8, 9）が含まれているため除外されています。

条件となる数字リストを変更すれば、さまざまなフィルタリング要件に柔軟に対応できる汎用性の高いテクニックです。