Python学习笔记—itertools模块

这篇是看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')

这个模块函数有好多，有好多敲了一遍忘了保存，懒得再敲了，但也记得差不多了，所以就这样吧