生成器
这种一边循环一边计算的机制,称为生成器(Generator)。
L = [x * x for x in range(10)] g = (x * x for x in range(10))
创建L和g的区别仅在于最外层的[]和(),L是一个list,而g是一个generator。
可以通过generator的next()方法,打印出来里面的元素
generator保存的是算法,每次调用next(),就计算出下一个元素的值,直到计算到最后一个元素,没有更多的元素时,抛出StopIteration的错误。
上面这种不断调用next()方法实在是太变态了,正确的方法是使用for循环,因为generator也是可迭代对象:
g = (x * x for x in range(10)) for n in g: ... print n
斐波拉契数列用列表生成式写不出来,但是,用函数把它打印出来却很容易:
def fib(max): n, a, b = 0, 0, 1 while n < max: print b a, b = b, a + b n = n + 1
要把fib函数变成generator,只需要把print b改为yield b就可以了:
def fib(max): n, a, b = 0, 0, 1 while n < max: yield b a, b = b, a + b n = n + 1
如果一个函数定义中包含yield关键字,那么这个函数就不再是一个普通函数,而是一个generator:
在每次调用next()的时候执行,遇到yield语句返回,再次执行时从上次返回的yield语句处继续执行。
yield相当于return。所以不在交互式命令行中这个返回被你丢弃了
generator是非常强大的工具,在Python中,可以简单地把列表生成式改成generator,也可以通过函数实现复杂逻辑的generator。
要理解generator的工作原理,它是在for循环的过程中不断计算出下一个元素,并在适当的条件结束for循环。对于函数改成的generator来说,遇到return语句或者执行到函数体最后一行语句,就是结束generator的指令,for循环随之结束。
其中注意:
a,b=b,a+b 相当于:
temp=b #先保存b的原值
b=a+b #赋b新值
a=temp #将b的原值赋予a
a, b = b, a + b 就是分别把等号右边两个值赋予等号左边的两个值。相当于 a=b b=a+b 的缩写。
高阶函数
abs(-10)是函数调用,而abs是函数本身。
把函数本身赋值给变量,那么变量就指向函数
函数名其实就是指向函数的变量!
函数的参数能接收变量,那么一个函数就可以接收另一个函数作为参数,这种函数就称之为高阶函数。
f = abs def add(x, y, f): return f(x) + f(y)
map 映射/reduce 归约
map将传入的函数依次作用到序列的每个元素
map()作为高阶函数,事实上它把运算规则抽象了,因此,我们不但可以计算简单的f(x)=x2,还可以计算任意复杂的函数,如下:
list(map(str, [1, 2, 3, 4, 5, 6, 7, 8, 9]))
结果是['1', '2', '3', '4', '5', '6', '7', '8', '9']
reduce的用法。reduce把一个函数作用在一个序列[x1, x2, x3, ...]上,这个函数必须接收两个参数,reduce把结果继续和序列的下一个元素做累积计算,其效果就是:
reduce(f, [x1, x2, x3, x4]) = f(f(f(x1, x2), x3), x4)
要把序列[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
下面是最难的例子(我是写不出来了, 只能背了)
到字符串str也是一个序列,对上面的例子稍加改动,配合map(),我们就可以写出把str转换为int的函数:
from functools import reduce
def str2int(s): def fn(x, y): return x * 10 + y def char2num(s): return {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}[s] return reduce(fn, map(char2num, s))
还可以用lambda函数进一步简化成(需要再看一次)
from functools import reduce
def char2num(s): return {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}[s]
def str2int(s): return reduce(lambda x, y: x * 10 + y, map(char2num, s))
练习题1(还需要再看)
利用map()函数,把用户输入的不规范的英文名字,变为首字母大写,其他小写的规范名字。输入:['adam', 'LISA', 'barT'],输出:['Adam', 'Lisa', 'Bart']。
def fn(name): return name.title()
map(fn, ['adam', 'LISA', 'barT'] )
s = [''adam', 'LISA', 'barT''] def string_good(s): return s[0].upper()+s[1:].lower()
print map(string_good,s)
练习题2
Python提供的sum()函数可以接受一个list并求和,请编写一个prod()函数,可以接受一个list并利用reduce()求积。
from functools import reduce def prod(L): return reduce(lambda x, y: x * y, L)
L = list(range(1, 11))
print(prod(L)) print('3 * 5 * 7 * 9 =', prod([3, 5, 7, 9]))
第二种做法:
from functools import reduce
def prod(L): def multi(x, y): return x * y return reduce(multi, 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): pass
print('str2float(\'123.456\') =', str2float('123.456'))
上面是模板,下面是答案
from functools import reduce
def str2float(s): XiaoShu = len(s) - s.find('.') - 1 # 计算小数位数 s = s.replace('.', '') # 删除小数点 def char2num(s): return {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}[s] num = reduce(lambda x, y: 10 * x + y, map(char2num, s)) # 转换成不带小数的整数 num = num / (10 ** XiaoShu) # 移动小数点 return num
print('str2float(\'123.456\') =', str2float('123.456'))
网友评论