Python 内置模块itertools的使用

Python的内建模块itertools提供了非常有用的用于操作迭代对象的函数，返回的是迭代器对象(实现__next__函数的类)。
 
1、itertools提供的几个“无限”迭代器：
(1)、count函数
import itertools
natuals = itertools.count(1)
for n in natuals:
print n
count()会创建一个无限的迭代器，所以上述代码会打印出自然数序列，根本停不下来，只能按Ctrl+C退出
i= 0
foritem in itertools.count(100):
if i>10:
break
print item,
i = i+1
功能：从100开始数10个数，cout返回一个无界的迭代器，小菜鸟我不会控制了，就引入了一个计数I，让它计数10次。。
输出：100 101 102 103 104105 106 107 108 109 110
 
(2)、cycle()会把传入的一个序列(字符串也是序列的一种)无限重复下去：
import itertools
listone= ['a','b','c']
listtwo= ['11','22','abc']

foritem in itertools.cycle(listone):
print item，
打印出a b c a b c a b c ab c a b c a b c a b c a b c a b c...
功能：从列表中取元素，到列表尾后再从头取...
无限循环，因为cycle生成的是一个无界的迭代器
(3)、repeat()负责把一个元素无限重复下去，不过如果提供第二个参数就可以限定重复次数：
ns = itertools.repeat('A', 10)
for n in ns:
print n 打印10次'A'

import itertools
listone = ['a','b','c']
for item initertools.repeat(listone,3):
print item,
2、其他的迭代器操作函数
(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)、ifilter的使用

ifilter(fun,iterator)

返回一个可以让fun返回True的迭代器

import itertools
listone = ['a','b','c']
listtwo = ['11','22','abc']

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

(3)、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中的元素每个加了５后返回给迭代器

 

(5)、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

 

(4)、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的组合