python itertools的使用(转)

1.　chain的使用

import itertools  
listone = ['a','b','c']  
listtwo = ['11','22','abc']  
for item in  itertools.chain(listone,listtwo):  
    print item  

输出: a b c 11 22 abc

2.　count的使用

i = 0  
for item in itertools.count(100):  
    if i>10:  
        break  
    print item,  
    i = i+1  
功能：从100开始数10个数，cout返回一个无界的迭代器，如果引入一个计数I，可以让它计数10次。。
输出：100 101 102 103 104 105 106 107 108 109 110

3.cycle的使用

import itertools  
listone = ['a','b','c']  
listtwo = ['11','22','abc']  
  
for item in itertools.cycle(listone):  
    print item，  
功能：从列表中取元素，到列表尾后再从头取...
无限循环，因为cycle生成的是一个无界的失代器
输出:a b c a b c a b c a b c a b c a b c a b c a b c a b c...

4.ifilter的使用，ifilter(fun,iterator)返回一个可以让fun返回True的迭代器

import itertools  

def funLargeFive(x):  
    if x > 5:  
        return True  
      
for item in itertools.ifilter(funLargeFive,range(-10,10)):  
    print item,  
结果：6 7 8 9

5.　imap的使用，imap(fun,iterator)返回一个迭代器，对iterator中的每个项目调用fun

import itertools  

listthree = [1,2,3]  
def funAddFive(x):  
    return x + 5  
for item in itertools.imap(funAddFive,listthree):  
    print item,  
返回：6　7　8　　对listthree中的元素每个加了５后返回给迭代器

6.islice的使用，islice()(seq, [start,] stop [, step])

import itertools  
listone = ['a','b','c']  
listtwo = ['11','22','abc']  
listthree = listone + listtwo  
for item in itertools.islice(listthree,3,5):  
    print item,  
功能：返回迭代器，其中的项目来自　将seq，从start开始,到stop结束，以step步长切割后
打印出：11 22

7.izip的使用，izip(*iterator)

import itertools  
listone = ['a','b','c']  
listtwo = ['11','22','abc']  
listthree = listone + listtwo  
for item in itertools.izip(listone,listtwo):  
    print item,  
结果：('a', '11') ('b', '22') ('c', 'abc')
功能：返回迭代器，项目是元组，元组来自*iterator的组合

8. repeate，repeate(elem [,n])

import itertools  
listone = ['a','b','c']  
for item in itertools.repeat(listone,3):  
    print item,  

结果：['a', 'b', 'c'] ['a', 'b', 'c'] ['a', 'b', 'c']