python 高阶函数 (二)--filter

filter()函数

用于过滤序列，从一个序列中筛出符合条件的元素，接收一个函数和一个序列

---

filter()函数返回的是一个Iterator，也就是一个惰性序列，所以要强迫filter()完成计算结果，需要用 list()函数 获得所有结果并返回list

---

删掉偶数，只保留奇数

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


if __name__ == '__main__':
    a = list(filter(is_odd, [1, 2, 34, 4, 23, 343, 234, 33, 24, 124, 5465, 464]))
    print(a)   # [1, 23, 343, 33, 5465]
    b = list(filter(not_empty, ['A', '', 'B', 'c', None, '   ']))
    print(b)   # ['A', 'B', 'c']

删除空字符串

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


if __name__ == '__main__':
    b = list(filter(not_empty, ['A', '', 'B', 'c', None, '   ']))
    print(b)   # ['A', 'B', 'c']

求素数

# 构造一个从3开始的奇数序列
def _odd_iter():
    n = 1
    while True:
        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)


if __name__ == '__main__':
    for n in primes():
        if n < 1000:
            print(n)
        else:
            break

练习题
利用filter()筛选出回数，回数是指从左向右读和从右向左读都是一样的数，例如12321，909

# 构造一个从3开始的奇数序列
def is_palindrome(n):
    m = str(n)
    return m == m[::-1]


if __name__ == '__main__':
    output = filter(is_palindrome, range(1, 1000))
    print('1~1000:', list(output))
    if list(filter(is_palindrome, range(1, 200))) == [1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 22, 33, 44, 55, 66, 77, 88, 99,
                                                      101, 111, 121, 131, 141, 151, 161, 171, 181, 191]:
        print('测试成功!')
    else:
        print('测试失败!')

方法二：用一行代码也可以实现

c = list(filter(lambda n : str(n) == str(n)[::-1], range(1, 100)))

方法三：

def is_palindrome(n):
    s = str(n)
    h = list(range((len(s) // 2)))
    for i in h:
        if s[i] != s[-(i+1)]:
            return False
    return True