Python学习笔记|二十一.map、reduce和filter

高阶函数之map/reduce

Python内建了map()``reduce()两个函数。

map（）函数

map()接受两个参数：

1. 一个是函数
2. 一个是interable

功能：map将传入的函数依次作用到序列的每个元素上面，然后将结果作为新的Interator返回。

例子：

 >>> def f(x):
...     return x * x
...
>>> r = map(f, [1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> list(r)
[1, 4, 9, 16, 25, 36, 49, 64, 81]

该例子功能是：将平方函数，依次作用到列表的每个元素上面，然后返回新的列表

>>> list(map(str, [1, 2, 3, 4, 5, 6, 7, 8, 9]))
['1', '2', '3', '4', '5', '6', '7', '8', '9']

reduce()函数

两个参数：

1. 函数
2. 列表

功能：reduce将函数作用在一个序列上面，将上面计算结果继续和下一个元素做累计计算。

例如：序列求和，序列重组

若Python未提供int（）函数，则可以自己写一个将字符串转化额外i整数的函数。

高阶函数之filter

filter()函数功能：用于过滤序列

参数：

1. 一个函数
2. 一个序列

功能：filter将每个函数作用于序列的每个元素，通过返回True或者False，来决定是否丢弃该元素

例子：

在一个list中，删掉偶数，只保留奇数，可以这么写：

def is_odd(n):
    return n % 2 == 1

list(filter(is_odd, [1, 2, 4, 5, 6, 9, 10, 15]))
# 结果: [1, 5, 9, 15]

把一个序列中的空字符串删掉，可以这么写：

def not_empty(s):
    return s and s.strip()

list(filter(not_empty, ['A', '', 'B', None, 'C', '  ']))
# 结果: ['A', 'B', 'C']

该函数返回的是一个惰性序列，所以要用list（）强制获得所有结果并返回ist。

用filter 求素数

• 埃氏筛法

1. 首先，列出从2开始的所有自然数，构造一个序列：

2. 取序列的第一个数2，它一定是素数，然后用2把序列的2的倍数筛掉：

3. 取新序列的第一个数3，它一定是素数，然后用3把序列的3的倍数筛掉：

4. 取新序列的第一个数5，然后用5把序列的5的倍数筛掉：

5. 不断筛下去，就可以得到所有的素数。

构造一个从3开始的奇数序列：这是一个生成器，并且是一个无限序列。

def _odditer():
    n = 1
    while True:
        n = n + 2
        yield n

定义一个筛选函数：

def _not_divisible(n):
    return lambda x: x % n > 0

定义一个生成器，不断返回下一个素数：

def primes():
    yield 2
    it = _odd_iter() # 初始序列
    while True:
        n = next(it) # 返回序列的第一个数
        yield n
        it = filter(_not_divisible(n), it) # 构造新序列

设置一个退出循环的条件：

# 打印1000以内的素数:
for n in primes():
    if n < 1000:
        print(n)
    else:
        break

小结

filter()的作用是从一个序列中筛出符合条件的元素。由于filter()使用了惰性计算，所以只有在取filter()结果的时候，才会真正筛选并每次返回下一个筛出的元素。