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')

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

上一篇:Java 语言基础 (初识Java语言, 变量和数据类型, 运算符, 流程控制语句, 数组)


下一篇:Python之路【第七篇】:初识Socket