前回の記事がmap関数の話だったので、今回は使い方のよく似たfilter関数です。
ドキュメントはこちら。
組み込み関数
基本的な構文は以下の形で、iterable の各要素に functionを適用して、
結果が新なものだけを取り出せます。
filter(function, iterable)
map関数と同様に、戻り値はリストではなくイテレーターになるので最初は少し戸惑いました。
試しに、整数のうち偶数だけ抽出するフィルターを書いてみます。
def f(n):
return n % 2 == 0
m = filter(f, range(10))
print(m)
# <filter object at 0x114d178d0>
print(list(m))
# [0, 2, 4, 6, 8]
print(list(m))
# []
細かい説明は mapの記事に書いた通りなのですが、
イテレーターを使うために、リストに変換するなり、nextで取り出すなりする必要があり、
一度取り出すともう一回listに変換しようとしても空のリストしか返ってこなくなります。
なお、内包表記でもほぼ同じ処理が実装でき、自分はこちらを使うことが多いです。
[x for x in range(10) if f(x)]
[0, 2, 4, 6, 8]