Python 的reduce()函數

reduce()函數也是Python內置的一個高階函數。reduce()函數接收的參數和 map()類似,一個函數 f,一個list,但行爲和 map()不同,reduce()傳入的函數 f 必須接收兩個參數,reduce()對list的每個元素反覆調用函數f,並返回最終結果值。

例如,編寫一個f函數,接收x和y,返回x和y的和:

def f(x, y):
    return x + y

 調用 reduce(f, [1, 3, 5, 7, 9])時,reduce函數將做如下計算:

先計算頭兩個元素:f(1, 3),結果爲4;
再把結果和第3個元素計算:f(4, 5),結果爲9;
再把結果和第4個元素計算:f(9, 7),結果爲16;
再把結果和第5個元素計算:f(16, 9),結果爲25;
由於沒有更多的元素了,計算結束,返回結果25。

 上述計算實際上是對 list 的所有元素求和。雖然Python內置了求和函數sum(),但是,利用reduce()求和也很簡單。

reduce()還可以接收第3個可選參數,作爲計算的初始值。如果把初始值設爲100,計算:

reduce(f, [1, 3, 5, 7, 9], 100)

 結果將變爲125,因爲第一輪計算是:

計算初始值和第一個元素:f(100, 1),結果爲101

實例1. Python內置了求和函數sum(),但沒有求積的函數,請利用recude()來求積:

from functools import reduce  #從functools模塊引入reduce()函數
def prod(x, y):
    return x * y

print (reduce(prod, [2, 4, 5, 7, 12]))
#輸出結果
3360

注:http://www.php.cn/python-tutorials-364636.html  關於functools模塊的詳細介紹

實例2. 考慮到字符串 str 也是一個序列,配合 map() 可以寫出把 str 裝換爲 int 的函數:

from functools import reduce
def str2int(s):
    def fn(x,  y):
        return x * 10 + y
    def char2num(s):
        return {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6,'7': 7, '8': 8, '9': 9}[s]
    return reduce (fn, map(char2num, s))
print(str2int('12345'))
#輸出結果
12345

也就是說,假設 Python 沒有提供 int()函數,你完全可以自己寫一個把字符串轉化爲整數的函數,而且只需要幾行代碼!
 

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