Python map和reduce函数用法示例

先看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

频道导航

Python map和reduce函数用法示例

您可能感兴趣的文章: