2018.2.2
1.列表生成式(List Comprehensions)
要生成list[1,2,3,4,5], 可以用list(range(1,6))
>>> list(range(1, 11))
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
要生成list[1*1,2*2,3*3,4*4,……,10*10],可以:
>>> [x * x for x in range(1, 11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
for循环后还可加入if判断语句,例如可筛选仅偶数的平方:
>>> [x * x for x in range(1, 11) if x % 2 == 0]
[4, 16, 36, 64, 100]
还可使用两层循环,生成全排列:
>>> [m + n for m in 'ABC' for n in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']
2.生成器(generator)
当需要创建一个巨大的列表但只需访问前面极少的几个元素,并且列表元素可以按照某种算法推算出来时,为了节省存储空间,我们可以在循环的过程中不断推算后续的元素,这种一边循环一边计算的机制,就叫做
生成器:generator。
要创建一个generator,只需将列表生成式的 [ ] 改成 ( ) 即可:
>>> L = [x * x for x in range(10)]
>>> L
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
>>> g = (x * x for x in range(10))
>>> g
<generator object <genexpr> at 0x1022ef630>
而要打印generator中的元素,有两种方法。第一种是通过next()函数,获得下一个返回值:
>>> next(g)
0
>>> next(g)
1
>>> next(g)
4
>>> next(g)
9
>>> next(g)
16
>>> next(g)
25
>>> next(g)
36
>>> next(g)
49
>>> next(g)
64
>>> next(g)
81
>>> next(g)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
StopIteration
此方法繁琐,不常用,常用的方法是第二种——使用for循环调用:
>>> g = (x * x for x in range(10))
>>> for n in g:
... print(n)
...
0
1
4
9
16
25
36
49
64
81
此外,如果推算的算法比较复杂,还可用函数实现。以斐波那契数列为例,其函数形式为:
def fib(max):
n, a, b = 0, 0, 1
while n < max:
print(b)
a, b = b, a + b
n = n + 1
return 'done'
而只要把print(b) 改为 yield b,便将函数改为了一个generator:
def fib(max):
n, a, b = 0, 0, 1
while n < max:
yield b
a, b = b, a + b
n = n + 1
return 'done'
generator和函数的执行流程不一样。函数是顺序执行,遇到
return
语句或者最后一行函数语句就返回。而变成generator的函数,在每次调用next()
的时候执行,遇到yield
语句返回,再次执行时从上次返回的yield
语句处继续执行。带你用的方法同样有两种,next方法不常用,主要记住for循环调用方法:
>>> for n in fib(6):
... print(n)
...
1
1
2
3
5
8
但是用
for
循环调用generator时,发现拿不到generator的return
语句的返回值。如果想要拿到返回值,必须捕获StopIteration
错误,返回值包含在StopIteration
的value
中:
>>> g = fib(6)
>>> while True:
... try:
... x = next(g)
... print('g:', x)
... except StopIteration as e:
... print('Generator return value:', e.value)
... break
...
g: 1
g: 1
g: 2
g: 3
g: 5
g: 8
Generator return value: done
2018.2.3
1.高级函数
变量可以指向函数;
函数的参数可以接收变量;
因此,一个函数可接收另一个函数作为参数,这种函数就叫做高级函数。
f = abs #内置求绝对值函数
def add(x, y, f):
return f(x) + f(y)