当前位置 博文首页 > Python map和reduce函数用法示例

    Python map和reduce函数用法示例

    作者:admin 时间:2021-07-25 10:48

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

    jsjbwy
    下一篇:没有了