Python函数式编程——返回函数

高阶函数除了可以接受函数作为参数外，还可以把函数作为结果值返回。

看代码：

# -*- coding: utf-8 -*-
# @File  : 返回函数的高阶函数.py
# @author: Flyme awei 
# @email : Flymeawei@163.com
# @Time  : 2022/8/21 14:48


def sum_fun(*args):

    def add_fun():
        s = 0
        for i in args:
            s += i
        return s
    return add_fun


f = sum_fun(1, 2, 3, 4, 5, 6, 7, 8, 9)
add = f()
print(add)  # 45

当我们调用 sum_fun 时，返回的并不是求和结果，而是求和函数 add_fun , 将其赋值给f，当我们在调f函数时才返回求和结果s。

用filter函数来计算素数

  用 filter来计算素数其中一个方法是埃氏筛法。
  给出要筛数值的范围 n，找出以内的素数。先用 2去筛，即把 2留下，把 2的倍数剔除掉；再用下一个质数，也就是 3筛，把 3留下，把 3的倍数剔除掉；接下去用下一个质数 5筛，把 5留下，把 5的倍数剔除掉；不断重复下去......

用Python高阶函数来实现这个算法：

1. 我们先写一个生成器构造一个从3开始的无限奇数序列，首先排除偶数。

def odd_num():  # 奇数生成器函数
    n = 1
    while True:
        n += 2
        yield n

1. 写一个筛选的函数，这里使用了匿名函数，返回判断是否为可整除数

def un_divisible(n):  # 判断是否为可整除数
    return lambda x: x % n > 0

1. 使用filter来过滤，不断返回素数的生成迭代

def primes():  # 素数生成器函数
    yield 2
    it = odd_num()
    while True:
        n = next(it)
        yield n
        it = filter(un_divisible(n), it)  # 过滤出不可以整除的数

1. 判断素数方法就产生了，这里需要手动结束一下

for i in primes():  # 打印小于100的素数
    if i < 100:
        print(i)
    else:
        break  

高阶函数实现打印小于100的素数：

# -*- coding: utf-8 -*-
# @File  : 返回函数计算质数.py
# @author: Flyme awei 
# @email : Flymeawei@163.com
# @Time  : 2022/8/21 15:17


# 得到所有的质数，打印小于100的所有的质数
# 最小的质数是2， 质数:只能被1和它本身整除的数
# 思路:先得到所有大于1的奇数， --> 生成器，在把生成器中的所有元素过滤去掉: 那些可以被小于元素本身的质数整除的数


# 1.得到所有的大于1奇数的生
def odd_num():
    n = 1
    while True:
        n += 2
        yield n


def un_divisible(n):  # 判断是否能够整除的函数，n代表从生成器中拿到的一个大于1的质数
    return lambda x: x % n > 0  # x是某一个奇数，n:小于当前x的一个质数


# 2.创建一个质数的生成器，最小的质数是2
def primes():
    yield 2
    n = odd_num()  # g为大于1的奇数生成器
    while True:
        x = next(n)
        g = filter(un_divisible(n), n)
        yield x


for i in primes():
    if i < 100:
        print(i, end=' ')
    else:
        break

第一段代码生成了以3开始的奇数序列
第二段代码自定义过滤函数，包含匿名函数，判断值的取余是否能被整除
第三段代码用来返回素数，这里先返回一个2为素数，因为偶数都被排除了所

这就是100以内的所有素数：