Python: map() and reduce()

Posted 2020-10-19 baobaoqin

map()函数接收两个参数，一个是函数，一个是序列，map将传入的函数依次作用到序列的每个元素，并把结果作为新的list返回。

举例说明，比如我们有一个函数f(x)=x²，要把这个函数作用在一个list [1, 2, 3, 4, 5, 6, 7, 8, 9]上，就可以用map()实现如下:

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

result:

[1, 4, 9, 16, 25, 36, 49, 64, 81]

reduce把一个函数作用在一个序列[x1, x2, x3...]上，这个函数必须接收两个参数，reduce把结果继续和序列的下一个元素做累积计算，其效果就是：

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

Practice:

【练习1】利用map()函数，把用户输入的不规范的英文名字，变为首字母大写，其他小写的规范名字。输入：[‘adam‘, ‘LISA‘, ‘barT‘]，输出：[‘Adam‘, ‘Lisa‘, ‘Bart‘]。

答：

def firstToUpper(x):
    s1 = x[:1].upper()
    s2 = x[1:].lower()
    s3 = s1 + s2
    return s3
map(firstToUpper,[‘aleN‘,‘TOM‘,‘hello‘])

解释： x[:1]: x string 的第一个字符（从 0 开始到 1 下标但是不包含1下标）； x[1:]: x string 的第二个字符一直到最后， 即从下标 1 开始到最后。。。

结果：

[‘Alen‘, ‘Tom‘, ‘Hello‘]

【练习2】Python提供的sum()函数可以接受一个list并求和，请编写一个prod()函数，可以接受一个list并利用reduce()求积。

def prod(x,y):
    return x+y
reduce(prod,[1,2,3,4,5,6,7,8100])

结果：

8128