Chinaunix首页 | 论坛 | 博客
  • 博客访问: 686114
  • 博文数量: 231
  • 博客积分: 3217
  • 博客等级: 中校
  • 技术积分: 2053
  • 用 户 组: 普通用户
  • 注册时间: 2011-07-04 12:01
文章分类

全部博文(231)

文章存档

2015年(1)

2013年(10)

2012年(92)

2011年(128)

分类: LINUX

2012-08-30 13:01:24

Python:itertools模块

itertools模块包含创建有效迭代器的函数,可以用各种方式对数据进行循环操作,此模块中的所有函数返回的迭代器都可以与for循环语句以及其他包含迭代器(如生成器和生成器表达式)的函数联合使用。

chain(iter1, iter2, ..., iterN):

给出一组迭代器(iter1, iter2, ..., iterN),此函数创建一个新迭代器来将所有的迭代器链接起来,返回的迭代器从iter1开始生成项,知道iter1被用完,然后从iter2生成项,这一过程会持续到iterN中所有的项都被用完。

View Code
复制代码
1 from itertools import chain
2 test = chain('AB', 'CDE', 'F')
3 for el in test:
4 print el
5
6 A
7 B
8 C
9 D
10 E
11 F
复制代码

chain.from_iterable(iterables):

一个备用链构造函数,其中的iterables是一个迭代变量,生成迭代序列,此操作的结果与以下生成器代码片段生成的结果相同:

View Code
复制代码
1 >>> def f(iterables):
2 for x in iterables:
3 for y in x:
4 yield y
5
6 >>> test = f('ABCDEF')
7 >>> test.next()
8 'A'
9
10
11 >>> from itertools import chain
12 >>> test = chain.from_iterable('ABCDEF')
13 >>> test.next()
14 'A'
复制代码


combinations(iterable, r):

创建一个迭代器,返回iterable中所有长度为r的子序列,返回的子序列中的项按输入iterable中的顺序排序:

View Code
复制代码
1 >>> from itertools import combinations
2 >>> test = combinations([1,2,3,4], 2)
3 >>> for el in test:
4 print el
5
6
7 (1, 2)
8 (1, 3)
9 (1, 4)
10 (2, 3)
11 (2, 4)
12 (3, 4)
复制代码

count([n]):

创建一个迭代器,生成从n开始的连续整数,如果忽略n,则从0开始计算(注意:此迭代器不支持长整数),如果超出了sys.maxint,计数器将溢出并继续从-sys.maxint-1开始计算。

cycle(iterable):

创建一个迭代器,对iterable中的元素反复执行循环操作,内部会生成iterable中的元素的一个副本,此副本用于返回循环中的重复项。

dropwhile(predicate, iterable):

创建一个迭代器,只要函数predicate(item)为True,就丢弃iterable中的项,如果predicate返回False,就会生成iterable中的项和所有后续项

View Code
复制代码
1 def dropwhile(predicate, iterable):
2 # dropwhile(lambda x: x<5, [1,4,6,4,1]) --> 6 4 1
3 iterable = iter(iterable)
4 for x in iterable:
5 if not predicate(x):
6 yield x
7 break
8 for x in iterable:
9 yield x
阅读(1222) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~