Python3中的map()、reduce()、filter()

Python3中的map()reduce()filter() 這3個一般是用於對序列進行操作的內置函數,它們經常需要與 匿名函數 lambda 聯合起來使用,我們今天就來學習下。

map()

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() 可以用於過濾序列,過濾掉不符合條件的元素,返回值也是一個迭代器,其使用語法如下:

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 點需要注意:

  • map 中必須傳入一個正常函數,而在 filter 函數中則可以傳正常函數或者None,當傳入None時,只返回可迭代對象中所有符合 true 的值
  • map 中支持傳多個可迭代對象,而在 filter 函數中則只能傳一個可迭代對象

reduce()

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 所有整數之和"""
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中的高階函數,它們最大的好處在於可以讓代碼更加簡潔,當然,如果不使用它們,我們也可以通過其他方式來實現。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章