Python的高级特性之切片、迭代、列表生成式、生成器

切片

切片就是获取一个list、tuple、字符串等的部分元素

 l = range(100)
 #取[0,5)元素
 print(l[:5]) #[0, 1, 2, 3, 4]
 #在[0,99]中每隔10个元素取一个
 print( l[::10]) #[0, 10, 20, 30, 40, 50, 60, 70, 80, 90]
 #取最后五个元素
 print( l[-5 : ]) #[95, 96, 97, 98, 99] 

 str = "www.genekang.com"
 #也可以用来截取字符串
 print( str[4:8]) #gene

迭代

如果给定一个list或tuple，我们可以通过for循环来遍历这个list或tuple，这种遍历我们成为迭代（Iteration）

 myList = ['a', 'b', 123, 'abc']

 #对一个list进行迭代

 for k in myList:

     print k

 #Python内置的enumerate函数可以把一个list变成索引-元素对

 # 0 a

 # 1 b

 # 2 123

 # 3 abc

 for i,k in enumerate(myList):

     print i,k

除了list之外，dict迭代的是key。如果要迭代value，可以用for value in d.itervalues()，如果要同时迭代key和value，可以用for k, v in d.iteritems()。

如何判断一个对象是可迭代对象呢？方法是通过collections模块的Iterable类型判断：

>>> from collections import Iterable
>>> isinstance('abc', Iterable) # str是否可迭代
True
>>> isinstance([1,2,3], Iterable) # list是否可迭代
True
>>> isinstance(123, Iterable) # 整数是否可迭代
False

列表生成式

列表生成式即List Comprehensions，是Python内置的非常简单却强大的可以用来创建list的生成式

 #计算1-10中偶数的平方

 ll =range(1,11)

 print [x*x for x in ll if x % 2 ==0 ]

 #还可以双层循环

 print [i + j for i in "ab" for j in "cd"] #['ac', 'ad', 'bc', 'bd']

 #列表生成式也可以使用两个变量来生成list

 d = {'x': 'A', 'y': 'B', 'z': 'C' }

 print [k + ':' + v for k,v in d.iteritems()] #['y:B', 'x:A', 'z:C']

生成器

在Python中，这种一边循环一边计算的机制，称为生成器（Generator），相当于java中的迭代器（iterator）

要创建一个generator，有很多种方法。第一种方法很简单，只要把一个列表生成式的[]改成()，就创建了一个generator
 

 用函数实现一个斐波那契数列

 #1 1 2 3 5 8 

 def fib(max): 

     n,a,b = 0,0,1 

     while(n<max): 

         print b 

         a , b = b , a+b 

         n = n+1 

仔细观察，可以看出，fib函数实际上是定义了斐波拉契数列的推算规则，可以从第一个元素开始，推算出后续任意的元素，这种逻辑其实非常类似generator。

也就是说，上面的函数和generator仅一步之遥。要把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

 for x in fib(6):

     print x