标签：
这篇是看wklken的《Python进阶-Itertools模块小结》 学习itertools模块的学习笔记
在看itertools中各函数的源代码时，刚开始还比较轻松，但后面看起来就比较费劲。。。
1、itertools.count(start=0,step=1)
此函数用来创建一个迭代器，生成从n开始的连续整数，如果忽略n，则从0开始计算
如果超出了sys.maxint，计数器将溢出并继续行-sys.maxint-1开始计算
定义：
def count(start=0, step=1): #count(10) --> 10, 11, 12, 13..... # count(2.5, 0.5)--> 2.5, 3.0, 3.5 .... n = start while True: yield n n += step
使用：
from itertools import * for i in izip(count(i), [‘a‘, ‘b‘, ‘c‘]): print i out: (1, ‘a‘) (2, ‘b‘) (3, ‘c‘)
2、itertools.cycle(iterable)
创建一个迭代器，对iterable中的元素反复执行循环操作，内部会生成iterable中的元素的一个副本， 次副本用于返回循环中的重复项
定义：
def cycle(iterable): # cycle(‘ABCD‘) --> A B C D A B C D .... saved = [] for element in iterable: yield element saved.append(element) while saved: for element in saved: yield element
使用：
from itertools import * i = 0 for item in cycle([‘a‘, ‘b‘, ‘c‘]): i += 1 if i == 6: break print (i, item) out: (1, ‘a‘) (2, ‘b‘) (3, ‘c‘) (4, ‘a‘) (5, ‘b‘)
3、itertools.repeat(object[, times])
创建一个迭代器，重复生成object， times (如果已提供) 指定重复计数， 如果未提供times， 将无尽返回该对象
定义：
def repeat(object, times=None): # repeat(10, 3) --> 10, 10, 10 if times is None: while True: yield object else: for i in xrange(time): yield object
使用：
from itertools import * for i in repeat(‘over-and-over‘, 3): print i out: over-and-over over-and-over over-and-over
4、itertools.chain(*iterables)
将多个迭代器作为参数，但只返回单个迭代器，它产生所有参数迭代器的内容，就好像他们来自于一个单一的序列。
定义：
def chain(*iterables): # chain(‘ABC‘, ‘DEF‘) --> A B C D E F for it in iterables: for element in it: yield element
使用：
from itertools import * for i in chain([1, 2, 3], [‘a‘, ‘b‘, ‘c‘]): print i out: 1 2 3 a b c
5、itertools.compress(data, selectors)
提供一个选择列表， 对原始数据进行筛选
定义：
def compress(data, selectors): # compress(‘ABCDEF‘, [1, 0, 1, 0, 1, 1]) --> A C E F return (d for d, s in izip(data, selectors) if s)
6、itertools.product(*iterables[, repeat])
笛卡尔积
创建一个迭代器，生成item1, item2等中的项目的笛卡尔积的元组， repeat是一个关键字参数，指定重复生成序列的次数。
def product(*args, **kwds): # product(‘ABCD‘, ‘xy‘) --> Ax, Ay, Bx, By, Cx, Cy, Dx, Dy # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111 pools = map(tuple, args) * kwds.get(repeat, 1) result = [[]] for pool in pools: result = [x+[y] for x in result for y in pool] for prod in result: yield tuple(prod)
import itertools a = (1, 2, 3) b = (‘a‘, ‘b‘, ‘c‘) c = itertools.product(a, b) for elem in c: print elem out: (1, ‘A‘) (2, ‘B‘) (3, ‘C‘) (2, ‘A‘) (2, ‘B‘) (2, ‘C‘) (3, ‘A‘) (3, ‘B‘) (3, ‘C‘)
这个模块函数有好多，有好多敲了一遍忘了保存，懒得再敲了，但也记得差不多了，所以就这样吧
标签：
原文地址：http://www.cnblogs.com/siriuswang/p/4633703.html
