Pythonのfilter()関数でリスト内のxより小さい値をすべて抽出する方法

数値のリスト nums と、もうひとつの数値 x が与えられたとします。このとき、nums の中から x より小さい値だけをフィルタリングして取り出す方法を解説します。

Pythonには filter() という組み込み関数があり、引数として関数を受け取り、その関数の条件に合う要素だけを抽出できます。これを使えば、簡潔なコードで目的の処理を実現できます。

問題の例

たとえば、入力が次のような場合を考えてみましょう。

nums = [1,5,8,3,6,9,12,77,55,36,2,5,6,12,87]
x = 50

この場合、50より小さい値だけが残るため、出力は次のようになります。

[1, 5, 8, 3, 6, 9, 12, 36, 2, 5, 6, 12]

解決の手順

• 引数 a を受け取る関数（またはラムダ式）を定義します。

• a < x であれば True を返し、そうでなければ False を返すようにします。

• filter() を使って、この関数の条件を満たす要素だけを nums から抽出します。

• filterオブジェクトはイテレータなので、list() でリストに変換して返します。

実装例

以下のコードは、ラムダ式を filter() に渡して処理を行うシンプルな実装です。

def solve(nums, x):
    left_items = filter(lambda a: a < x, nums)
    return list(left_items)

nums = [1,5,8,3,6,9,12,77,55,36,2,5,6,12,87]
x = 50
print(solve(nums, x))

入力

[1,5,8,3,6,9,12,77,55,36,2,5,6,12,87], 50

出力

[1, 5, 8, 3, 6, 9, 12, 36, 2, 5, 6, 12]

補足：リスト内包表記との比較

同じ処理は、リスト内包表記を使っても書けます。どちらも読みやすく効率的ですが、状況に応じて使い分けるとよいでしょう。

def solve(nums, x):
    return [a for a in nums if a < x]

filter() は遅延評価されるため、大量のデータを扱う場合でもメモリ効率が良いという利点があります。一方、リスト内包表記はPythonらしい直感的な書き方として広く好まれています。