python中itertools模組介紹---01

標籤：

itertools模組中包含了很多函數，這些函數最終都產生一個或多個迭代器，下面對這些函數進行介紹：

為了能夠使用itertools中的函數，需要將該模組匯入：

>>>from itertools import *

count(start=0,step=1):

原始碼為：

def count(start=0,step=1):    n=start    while True:        yield n        n+=step

從原始碼可以看出，count函數產生一個產生器，該產生器可以返回一個個數，預設是從0開始，每次增加1.例如：

>>>a=count(2,3)>>>a.next()2>>>a.next()5>>>a.next()8

當然，start和step也可以是小數。如果超出了sys.maxint，計數器將溢出，並繼續聰哥-sys.maxint-1開始計算。

cycle(iterable):

原始碼為：

def cycle(iterable):    saved=[]    for element in iterable:        yield element        saved.append(element)    while True:        for element in saved:            yield element

從原始碼可以看出，cycle函數建立了一個列表，然後將iterable中的元素儲存進去，最後無限返回列表中的元素。因此，cycle函數的作用是建立一個產生器，該產生器無限地返回參數中的元素，例如：

>>>a=cycle([1,2,3,4])>>>a.next()1>>>a.next()2>>>a.next()3>>>a.next()4>>>a.next()1

repeat(object[,times]):

原始碼如下：

def repeat(object,times=None):    if times is None:        while True:            yield object    else:        for i in xrange(times):            yield object

當times沒有被指定時，repeat無限重複，返回原對象。當times指定後，將重複times次返回該對象。例如：

>>>a=repeat(‘abc‘,2)>>>a.next()‘abc‘>>>a.next()‘abc‘>>>a.next()StopIteration異常

python中itertools模組介紹---01