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