Python map和reduce函数用法示例

程序员文章站 2022-04-21 08:48:13

...

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

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

复制代码代码如下:

>>> def a(x):
... return x * 2
...
>>> map(a, [1,2,3,4,5])
[2, 4, 6, 8, 10]

map传入的第一个参数a,即a函数，当然你也可以不用map函数实现这功能：

复制代码代码如下:

>>> list = []
>>> for i in [1, 2, 3, 4, 5]:
... list.append(a(i))
...
>>> print list
[2, 4, 6, 8, 10]

从代码量上来讲，map要精简很多，所以，map()作为高阶函数，事实上它把运算规则抽象了，因此，我们不但可以计算简单的a(x)=x＊2，还可以计算任意复杂的函数，比如，把这个list所有数字转为字符串：

复制代码代码如下:

>>> map(str,[1,2,3,4,5])
['1', '2', '3', '4', '5']
>>>

只需要一行代码，就搞定了。让我们再看和来自顾雪峰python教程的习题：利用map()函数，把用户输入的不规范的英文名字，变为首字母大写，其他小写的规范名字。输入：[‘adam', ‘LISA', ‘barT']，输出：[‘Adam', ‘Lisa', ‘Bart']。作为我个人来说，我可能会先将不规范的英文名全转换在小写然后再通过capitalize()函数，将首字母转换在写，代码如下：

复制代码代码如下:

>>> def caps(name):
... return name.capitalize()
...
>>> def lowers(name):
... return name.lower()
...
>>> map(caps, map(lowers,['adam', 'LISA', 'barT']))
['Adam', 'Lisa', 'Bart']

再看reduce的用法。reduce(function, sequence, starting_value)：对sequence中的item顺序迭代调用function，如果有starting_value，还可以作为初始值调用，例如可以用来对List求和：

复制代码代码如下:

>>> def add(x, y):
... return x + y
...
>>> reduce(add, [1, 3, 5, 7, 9])
25
>>> reduce(add, range(1, 11))
55
>>> reduce(add, range(1, 11),20)
75

当然求和运算可以直接用Python内建函数sum()，没必要动用reduce。但是如果要把序列[1,2,3,4,5,6,7]变换成整数1234567，reduce就可以派上用场：

复制代码代码如下:

>>> def fn(x, y):
... return x * 10 + y
...
>>> reduce(fn, [1,3,4,5,6,7])
134567