1. 程式人生 > >map、reduce函數

map、reduce函數

div 可選參數 元素 space 參數 clas 返回結果 一輪 lis

map函數:

接受一個函數 f 和一個 list 。格式:map( f , L),對L中的每個元素,進行f(x)的一個操作。

例如,對於list [1, 2, 3, 4, 5, 6, 7, 8, 9],如果希望把list的每個元素都作平方,就可以用map()函數:

def f(x):
    return x*x
print map(f, [1, 2, 3, 4, 5, 6, 7, 8, 9])

此時,結果就為:[1, 4, 9, 10, 25, 36, 49, 64, 81]

reduce函數:

接受一個函數 f 和一個 list 。格式:map( f , L)。與map不同的是,reduce中 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。

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

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

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

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

map、reduce函數