python-切片 迭代 生成器

1 切片操作

>>> L
['aaa', 'bbb', 'ccc', 'ddd']
>>> L[0:3]
['aaa', 'bbb', 'ccc']
>>> L[::5]
[0, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80, 85, 90, 95]
0-100的所有数字，每5个取一个

2 迭代操作 只要有可迭代对象就可以迭代

>>> d = {'a':1, 'b':2, 'c':3}
>>> for key in d:
...     print key
...
a
c
b

对字典进行迭代 默认迭代键 如果要迭代值，可以选用下面代码

>>> d
{'a': 1, 'c': 3, 'b': 2}
>>> for values in d.itervalues():
...     print values
...
1
3
2

d.itervalues()返回一个字典值的可迭代对象

同时迭代键和值：

>>> for k, v in d.iteritems():
...     print k, v
...
a 1
c 3
b 2

2.1 判断可迭代对象

>>> from collections import Iterable
>>> isinstance(d.iteritems(), Iterable)
True

2.3 enumerate函数可以把一个list变成索引-元素对，这样就可以在for循环中同时迭代索引和元素本身

>>> for i, v in enumerate(['a', 'b', 'c']):
...     print i, v
...
0 a
1 b
2 c

2.4 列表生成式

>>> [x*x for x in range(1, 11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
需要生成的x*x元素放在前面

3 生成器

生成列表的过程中是根据计算公式，边计算边生成，而不是一次生成

3.1 只要把一个列表生成式的[]改成()，就创建了一个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 0x7f87935774b0>

通过生成器的next()方法可以很方便的生成后续元素，全部打印需要使用for循环变量

>>> g.next()
0
>>> g.next()
1
>>> g.next()
4
>>> g.next()
9
>>> for n in g:
...     print n
...
16
25
36
49
64
81

3.2 通过yield也可以使用生成器函数

如下通过yield生成一个生成器函数

>>> def fib(max):
...     n, a, b = 0, 0, 1
...     while(n < max):
...             yield b
...             a, b = a, a + b
...             n = n + 1
...
>>> fib(6)
<generator object fib at 0x7f8793577aa0>

生成器函数和普通函数执行不同，生成器函数每往生成一次，就会从上一次yield处执行

代码实例如下：

>>> def odd():
...     print 'step 1'
...     yield 1
...     print 'step 2'
...     yield 2
...     print 'step 3'
...     yield 3
...
>>> o = odd()
>>> o.next()
step 1
1
>>> o.next()
step 2
2
>>> o.next()
step 3
3