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