Python3中的map()、reduce()、filter() 這3個一般是用於對序列進行操作的內置函數,它們經常需要與 匿名函數 lambda 聯合起來使用,我們今天就來學習下。
map() 可以用於在函數中對指定序列做映射,返回值是一個迭代器,其使用語法如下:
map(function, *iterables)
上面的第一個參數 function 指一個函數,第二個參數 iterable 指一個或多個可迭代對象,在執行過程中,會對可迭代對象中的每一個元素調用 function 函數做計算,最後得到一個新的迭代器對象,而這個新的迭代器對象,會包含有每次調用 function 函數的返回值。
只傳入一個可迭代對象
"""計算列表中每個元素的三次方"""
def demo_map(x):
return x ** 3
nums = [1, 2, 3, 4, 5]
print(list(map(demo_map, nums))) # 輸出:[1, 8, 27, 64, 125]
# 使用匿名函數
print(list(map(lambda x: x ** 3, nums))) # 輸出:[1, 8, 27, 64, 125]
傳入多個可迭代對象
"""計算3個列表中對應下標元素的和"""
def demo_map(x, y, z):
return x + y + z
nums1 = [1, 2, 3, 4, 5]
nums2 = [11, 22, 33, 44, 55]
nums3 = [100, 200, 300, 400, 500]
print(list(map(demo_map, nums1, nums2, nums3))) # 輸出:[112, 224, 336, 448, 560]
# 使用匿名函數
print(list(map(lambda x, y, z: x + y + z, nums1, nums2, nums3))) # 輸出:[112, 224, 336, 448, 560]
filter() 可以用於過濾序列,過濾掉不符合條件的元素,返回值也是一個迭代器,其使用語法如下:
filter(function or None, iterable)
和 map() 函數類似,上面的第一個參數 function 指一個函數,第二個參數 iterable 指一個可迭代對象,執行後會得到一個包含每次調用 function 函數返回值的迭代器。
"""找出從 -5 到 5 中能被 4 整除的所有整數"""
def demo_filter(x):
return x % 4 == 0
nums = range(-5, 6)
print(list(nums)) # 輸出:[-5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5]
# 傳入None,只返回true的值(0是False,所以被過濾掉了)
print(list(filter(None, nums))) # 輸入:[-5, -4, -3, -2, -1, 1, 2, 3, 4, 5]
# 傳入正常函數,過濾出 nums 中能被 4 整除的整數
print(list(filter(demo_filter, nums))) # 輸出:[-4, 0, 4]
# 使用匿名函數
print(list(filter(lambda x: x % 4 == 0, nums))) # 輸出:[-4, 0, 4]
針對 map() 和 filter() 函數, 這裡有 2 點需要注意:
reduce() 可以用於對參數序列中的元素進行累積,返回的是一個值。
在 Python3 中,reduce() 已被從全局名字空間裡移除了,如果想要使用它,那麼需通過引入 functools 模塊來調用 reduce() 函數,其使用語法如下:
from functools import reduce
reduce(function, sequence[, initial])
上面的第一個參數 function 指一個函數,並且該函數必須含有2個參數,第二個參數 sequence 指一個序列,第三個參數 initial 指初始值,默認是None。
例如存在函數:reduce(lambda x, y: x+y, [1, 2, 3, 4, 5]),它就相當於 ((((1+2)+3)+4)+5)。
"""計算 1-100 所有整數之和"""
''' 學習中遇到問題沒人解答?小編創建了一個Python學習交流群:711312441 尋找有志同道合的小伙伴,互幫互助,群裡還有不錯的視頻學習教程和PDF電子書! '''
from functools import reduce
def demo_reduce(x, y):
return x + y
nums = range(1, 101)
print(reduce(demo_reduce, nums)) # 輸出:5050
# 使用匿名函數
print(reduce(lambda x, y: x + y, nums)) # 輸出:5050
# 設置初始值為 1000
print(reduce(lambda x, y: x + y, nums, 1000)) # 輸出:6050
上面的 map()、reduce()、filter() 都是屬於Python3中的高階函數,它們最大的好處在於可以讓代碼更加簡潔,當然,如果不使用它們,我們也可以通過其他方式來實現。