允许把一个函数作为参数传入另一个参数
一、高阶函数
把函数看作是一个变量,也可以赋值
def add(x, y, f): return f(x) + f(y) print(add(-5, 6, abs)) # 其中f = abs就是计算绝对值
reduce把结果和下一个元素做计算
map返回list
map和reduce
利用
map()
函数,把用户输入的不规范的英文名字,变为首字母大写,其他小写的规范名字。输入:['adam', 'LISA', 'barT']
,输出:['Adam', 'Lisa', 'Bart']
:def normalize(name): return name.capitalize() print(map(normalize, ['adam', 'LISA', 'barT']))
如果要把序列
[1, 3, 5, 7, 9]
变换成整数13579
,reduce
就可以派上用场>>> from functools import reduce >>> def fn(x, y): ... return x * 10 + y ... >>> reduce(fn, [1, 3, 5, 7, 9]) 13579
Python提供的
sum()
函数可以接受一个list并求和,请编写一个prod()
函数,可以接受一个list并利用reduce()
求积:def ji(x, y): return x * y def prod(L): return reduce(ji, L) print('3 * 5 * 7 * 9 =', prod([3, 5, 7, 9]))
利用
map
和reduce
编写一个str2float
函数,把字符串'123.456'
转换成浮点数123.456
:from functools import reduce def str2float(s): DIGITS = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9} def I(x, y): return x * 10 + y def F(x, y): return x * 0.1 + y def char2num(s): return DIGITS[s] str1 = s.split('.')[0] str2 = s.split('.')[1][::-1] # 倒序输出 sum1 = map(char2num, str1) sum2 = map(char2num, str2) print(str2) return reduce(I, sum1) + reduce(F, sum2) * 0.1 print('str2float(\'123.456\') =', str2float('123.456'))
filter
Python内建的filter()
函数用于过滤序列
和map()
类似,filter()
也接收一个函数和一个序列。和map()
不同的是,filter()
把传入的函数依次作用于每个元素,然后根据返回值是True
还是False
决定保留还是丢弃该元素。
用filter求素数
首先,列出从
2
开始的所有自然数,构造一个序列:2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, ...
取序列的第一个数
2
,它一定是素数,然后用2
把序列的2
的倍数筛掉:3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, ...
取新序列的第一个数
3
,它一定是素数,然后用3
把序列的3
的倍数筛掉:5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, ...
取新序列的第一个数
5
,然后用5
把序列的5
的倍数筛掉:7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, ...
不断筛下去,就可以得到所有的素数。
用Python来实现这个算法,可以先构造一个从
3
开始的奇数序列:def _odd_iter(): 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) # 构造新序列
这个生成器先返回第一个素数
2
,然后,利用filter()
不断产生筛选后的新的序列。由于
primes()
也是一个无限序列,所以调用时需要设置一个退出循环的条件:# 打印1000以内的素数: for n in primes(): if n < 1000: print(n) else: break