Pythonのfilter()関数でデータをフィルタリングする方法

プログラミングをしていると、「小さい方のリストの各要素が、大きい方のリストに含まれているかどうかを確認したい」「ある条件に合致する要素だけを抽出したい」といった場面によく出会います。そんなときに活躍するのが、Python組み込みのfilter()関数です。

この記事では、filter()関数の基本的な構文から、実際のコード例、実行結果までをわかりやすく解説します。

filter()関数の基本構文

filter(function_name, sequence)

各引数の意味は次のとおりです。

• function_name：フィルタリングの条件を定義した関数の名前。各要素に対して呼び出され、True を返した要素だけが残ります。
• sequence：フィルタリング対象となるシーケンス（リスト、タプル、セットなど、その他のイテレータも可）。

filter()関数は、関数が True を返した要素のみを含むイテレータ（filterオブジェクト）を返します。

具体的な使用例

ここでは、月名が入ったリストを用意し、その中から特定の条件に合う月だけを抽出してみましょう。まず、抽出条件となる月名のリストを作成し、それをもとに判定を行う関数を定義します。そして、その関数を filter() に渡すことで、元のリストから該当する要素だけを取り出します。

# 元となる月のリスト
months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug']

# フィルタリング条件を判定する関数
def filterMonths(month):
    target_months = ['Apr', 'Jun', 'Aug', 'Oct']
    if month in target_months:
        return True
    else:
        return False

# filter()関数を適用
result = filter(filterMonths, months)

print('フィルタリングされた月 :')
for month in result:
    print(month)

このコードでは、filterMonths() 関数が各月名を受け取り、target_months リストに含まれているかどうかを判定しています。含まれていれば True を返すため、その要素だけが結果に残る仕組みです。

実行結果

上記のコードを実行すると、以下のような出力が得られます。

フィルタリングされた月 :
Apr
Jun
Aug

元のリスト8つの月の中から、条件に合致する「Apr」「Jun」「Aug」の3つだけが抽出されていることがわかります。

まとめ

filter()関数を使えば、forループで条件分岐を書くことなく、簡潔にリストやタプルなどの要素を絞り込むことができます。条件判定用の関数を別途定義するだけで再利用もしやすいため、データの抽出やクリーニングなど、さまざまな場面で役立つ便利な組み込み関数です。