Python杂记

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)