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
高阶函数来实现这个算法:
- 我们先写一个生成器构造一个从3开始的无限奇数序列,首先排除偶数。
def odd_num(): # 奇数生成器函数
n = 1
while True:
n += 2
yield n
- 写一个筛选的函数,这里使用了匿名函数,返回判断是否为可整除数
def un_divisible(n): # 判断是否为可整除数
return lambda x: x % n > 0
- 使用
filter
来过滤,不断返回素数的生成迭代
def primes(): # 素数生成器函数
yield 2
it = odd_num()
while True:
n = next(it)
yield n
it = filter(un_divisible(n), it) # 过滤出不可以整除的数
- 判断素数方法就产生了,这里需要手动结束一下
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以内的所有素数: