1. 程式人生 > >python中的map()函式和reduce()函式

python中的map()函式和reduce()函式

  • map()函式的用法:

    map(函式function_name, 可迭代物件Iterable)

map()函式接受兩個引數,一個是函式,一個是可迭代物件。map()函式將傳入的函式一次作用在序列的每一個元素,並把結果作為新的可迭代物件返回。

舉例說明,比如我們有一個函式f(x)=x2,要把這個函式作用在一個list [1, 2, 3, 4]上,就可以用map()實現如下:

def f(x):
    return x*x
r = map(f,[1,2,3,4])
print(list(r))
>>>
[1,4,9,16]

map()函式作為高階函式,把運算規則抽象了。我們可以一眼看明白上述程式碼的作用。

#把這個list中的所有數字轉化為字串
print(list(map(str, [1, 2, 3, 4, 5, 6, 7, 8, 9])))
>>>['1', '2', '3', '4', '5', '6', '7', '8', '9']
  • reduce()函式的用法
reduce(f, [x1, x2, x3, x4]) = f(f(f(x1, x2), x3), x4)
#將序列[1,2,3,4,5,6]變換成整數123456
from functools import reduce
def f(x,y):
    return 10*x + y
print(reduce(f,[1,2,3,4,5,6]))
>>>123456
from functools import  reduce
digits = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
def str2int(s):
    def f(x,y):
        return 10*x+y
    def char2num(s):
        return digits[s]
    return reduce(f,map(char2num,s))
print(str2int('123456'))
>>>123456

練習 1、利用map()函式,把使用者輸入的不規範的英文名字,變為首字母大寫,其他小寫的規範名字。輸入:[‘adam’, ‘LISA’, ‘barT’],輸出:[‘Adam’, ‘Lisa’, ‘Bart’]:

def normalize(name):
    def f(s):
        return s.capitalize()
    return map(f,name)
name = ['adam', 'LISA', 'barT']
print(list(normalize(name)))
>>>['Adam', 'Lisa', 'Bart']

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

from functools import reduce
def prod(L):
    def f(x,y):
        return x*y
    return reduce(f,L)
print('3 * 5 * 7 * 9 =', prod([3, 5, 7, 9]))
if prod([3, 5, 7, 9]) == 945:
    print('測試成功!')
else:
    print('測試失敗!')
    
>>>
3 * 5 * 7 * 9 = 945
測試成功!