Python中的map()函數和reduce()函數的用法

Python中的map()函數和reduce()函數的用法

這篇文章主要介紹了Python中的map()函數和reduce()函數的用法,代碼基於Python2.x版本,須要的朋友能夠參考下
 

Python內建了map()和reduce()函數。app

若是你讀過Google的那篇大名鼎鼎的論文「MapReduce: Simplified Data Processing on Large Clusters」,你就能大概明白map/reduce的概念。函數

咱們先看map。map()函數接收兩個參數,一個是函數,一個是序列,map將傳入的函數依次做用到序列的每一個元素,並把結果做爲新的list返回。spa

舉例說明,好比咱們有一個函數f(x)=x2,要把這個函數做用在一個list [1, 2, 3, 4, 5, 6, 7, 8, 9]上,就能夠用map()實現以下:.net

2015427113120966.png (341×245)


 

如今,咱們用Python代碼實現:code

?
1
2
3
4
5
>>> def f(x):
...   return x * x
...
>>> map (f, [ 1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 9 ])
[ 1 , 4 , 9 , 16 , 25 , 36 , 49 , 64 , 81 ]

map()傳入的第一個參數是f,即函數對象自己。htm

你可能會想,不須要map()函數,寫一個循環,也能夠計算出結果:對象

?
1
2
3
4
L = []
for n in [ 1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 9 ]:
   L.append(f(n))
print L

的確能夠,可是,從上面的循環代碼,能一眼看明白「把f(x)做用在list的每個元素並把結果生成一個新的list」嗎?ci

因此,map()做爲高階函數,事實上它把運算規則抽象了,所以,咱們不但能夠計算簡單的f(x)=x2,還能夠計算任意複雜的函數,好比,把這個list全部數字轉爲字符串:字符串

?
1
2
>>> map ( str , [ 1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 9 ])
[ '1' , '2' , '3' , '4' , '5' , '6' , '7' , '8' , '9' ]

只須要一行代碼。get

再看reduce的用法。reduce把一個函數做用在一個序列[x1, x2, x3...]上,這個函數必須接收兩個參數,reduce把結果繼續和序列的下一個元素作累積計算,其效果就是:

reduce(f, [x1, x2, x3, x4]) = f(f(f(x1, x2), x3), x4)

比方說對一個序列求和,就能夠用reduce實現:

?
1
2
3
4
5
>>> def add(x, y):
...   return x + y
...
>>> reduce (add, [ 1 , 3 , 5 , 7 , 9 ])
25

固然求和運算能夠直接用Python內建函數sum(),不必動用reduce。

可是若是要把序列[1, 3, 5, 7, 9]變換成整數13579,reduce就能夠派上用場:

?
1
2
3
4
5
>>> def fn(x, y):
...   return x * 10 + y
...
>>> reduce (fn, [ 1 , 3 , 5 , 7 , 9 ])
13579

這個例子自己沒多大用處,可是,若是考慮到字符串str也是一個序列,對上面的例子稍加改動,配合map(),咱們就能夠寫出把str轉換爲int的函數:

?
1
2
3
4
5
6
7
8
>>> 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]
...
>>> reduce (fn, map (char2num, '13579' ))
13579

整理成一個str2int的函數就是:

?
1
2
3
4
5
6
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))

還能夠用lambda函數進一步簡化成:

?
1
2
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]
?
1
2
def str2int(s):
   return reduce ( lambda x,y: x * 10 + y, map (char2num, s))

也就是說,假設Python沒有提供int()函數,你徹底能夠本身寫一個把字符串轉化爲整數的函數,並且只須要幾行代碼!

練習

利用map()函數,把用戶輸入的不規範的英文名字,變爲首字母大寫,其餘小寫的規範名字。輸入:['adam', 'LISA', 'barT'],輸出:['Adam', 'Lisa', 'Bart']。

Python提供的sum()函數能夠接受一個list並求和,請編寫一個prod()函數,能夠接受一個list並利用reduce()求積。

相關文章
相關標籤/搜索