1. 程式人生 > 實用技巧 >map函式和reduce函式

map函式和reduce函式

一、map函式

map 函式接收兩個引數,一個是函式,另一個是 Iterable,map 將傳入的函式依次作用到序列的每個元素,並把結果作為新的 Iterator 返回

例如:

>>> def f(x):
	return x*x

>>> l=[1,2,3,4,5,6,7,8,9]
>>> r=map(f,l)
>>> r
<map object at 0x0000028C820326A0>
>>> list(r)
[1, 4, 9, 16, 25, 36, 49, 64, 81]

  

二、reduce函式

reduce把一個函式作用在一個序列 [ x1,x2,x3......]上,這個函式必須有接收兩個引數,reduce把結果繼續和序列的下一個元素做累計計算

其效果為:

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

  

例如,對一個序列求和:

>>> from functools import reduce
>>> def add(x,y):
	return x+y

>>> reduce(add,[1,2,3,4,5])
15

  

三、相關練習題

1. 利用 map()函式,把使用者輸入的不規範的英文名字,變為首字母大寫,其他小寫的規範名字

def normalize(name):
    return name[0].upper()+name[1:].lower()

  

2.編寫一個 prod()函式,可以接受一個list並利用 reduce()求積

from functools import reduce
def prod(L):
    return reduce(lambda x,y:x*y,L)

  

3.利用 map和 reduce 編寫一個 str2float 函式,把字串'123.456'轉換成浮點數 123.456

from functools import reduce

def str2float(s):
    digits = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
    pointIndex=-1
    for i in range(0,len(s)):
        if ord(s[i]) == 46:
            pointIndex=i
            break
    if pointIndex >0:
        return reduce(lambda x,y:x*10+y,list(map(lambda z:digits[z],s[:pointIndex]+s[pointIndex+1:])))/(10**pointIndex)
    else:
        return reduce(lambda x,y:x*10+y,list(map(lambda z:digits[z],s)))