Python內建的filter()函數用于過濾序列。
和map()類似,filter()也接收一個函數和一個序列。和map()不同的是,filter()把傳入的函數依次作用于每個元素,然后根據返回值是True還是False決定保留還是丟棄該元素。
例如,在一個list中,刪掉偶數,只保留奇數,可以這么寫:
def is_odd(n): return n % 2 == 1 L1 = [1,2,3,4,5,6,7,8] l1 = list(filter(is_odd,L1)) print(l1)
結果:
[1, 3, 5, 7] [Finished in 0.2s]
可見用filter()這個高階函數,關鍵在于正確實現一個“篩選”函數。
注意到filter()函數返回的是一個Iterator,也就是一個惰性序列,所以要強迫filter()完成計算結果,需要用list()函數獲得所有結果并返回list。