今回は無名関数lambda（ラムダ式）のより実践的な例として、組み込み関数filter()とlambdaをあわせて使う方法を学びましょう。lambdaについてわからない場合はPython の無名関数 lambda（ラムダ式）の基本を参照ください。filter()関数については公式ドキュメントを一読するのが良いでしょう。
実際のソースコードで学ぶ
組み込み関数filter()と無名関数lambda（ラムダ式）とをあわせて使うと、リストから特定の要素だけを抽出することができます。
filter(function, iterable)
具体的にはiterable（第２引数）のうちfunction（第一引数）の返り値がFalseとなる要素がすべて除去され、Trueとなる要素だけを抽出することができます。説明よりも実際のソースコードの方がわかりやすいでしょう。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
list_1 = [1, 2, 3, 4, 5, 6, 7, 8, 9] list_2 = filter(lambda x: x % 2 == 0, list_1) # 偶数を抽出 # Python3でfilter()関数の仕様が変りました # Python2系では [2, 4, 6, 8] というリストが返ってきていましたが、3系では filter object が返ってきます print(list_2) # <filter object at XXXXXXXX> # Python3でリストとして取得するしたい場合 list_2 = list(filter(lambda x: x % 2 == 0, list_1)) print(list_2) # [2, 4, 6, 8] # tuple()関数を使うとタプルとして取得も可能 list_3 = tuple(filter(lambda x: x % 2 == 0, list_1)) print(list_3) # (2, 4, 6, 8) # 文字列でフィルターするサンプル list_s_1 = ['apple', 'grape', 'banana', 'orange'] list_s_2 = list(filter(lambda x: 'n' in x, list_s_1)) # 'n' を含む要素を抽出 print(list_s_2) # ['banana', 'orange'] |
for と if を使っても同様のことができますが、filter() と lambda を使うと1行で簡潔に書けることがわかります。簡潔に書けることも重要ですが、何よりこのfilter()とlambdaのあわせ技を知っていないと他の人が書いたソースコードを読めずに困るかもしれないので、頭に入れておきましょう。
注意事項
上記サンプルコード中のコメントにもあるように組み込み関数filter()はPython2系と3系で返り値が異なり、互換性がありません。2系のコードを3系に移植する際にもこの知識が役に立つでしょう。
コメントを残す