python itertools模块_Python标准库：itertools模块

来源：j_hao104

简介

官方描述：Functional tools for creating and using iterators.即用于创建高效迭代器的函数。

itertools.chain(*iterable)

将多个序列作为一个单独的序列返回。 例如：

importitertools

foreachinitertools.chain('i','love','python'):

printeach

输出：

i

l

o

v

e

p

y

t

h

o

n

itertools.combinations(iterable, r)

返回指定长度的”组合” 例如：

importitertools

foreachinitertools.combinations('abc',2):

printeach

输出：

('a','b')

('a','c')

('b','c')

itertools.combinations_with_replacement(iterable, r)

返回指定长度的“组合”，组合内元素可重复 例如：

importitertools

foreachinitertools.combinations_with_replacement('abc',2):

printeach

输出：

('a','a')

('a','b')

('a','c')

('b','b')

('b','c')

('c','c')

itertools.product(*iterable[,repeat])

返回指定长度的所有组合，可理解为笛卡尔乘积 例如：

importitertools

foreachinitertools.product('abc',repeat=2):

printeach

(‘a’, ‘a’) (‘a’, ‘b’) (‘a’, ‘c’) (‘b’, ‘a’) (‘b’, ‘b’) (‘b’, ‘c’) (‘c’, ‘a’) (‘c’, ‘b’) (‘c’, ‘c’)

itertools.premutations(iteravle[,r])

返回长度为r的排列 例如：

importitertools

forvalueinitertools.permutations('abc',2):

printvalue

输出：

('a','b')

('a','c')

('b','a')

('b','c')

('c','a')

('c','b')

itertools.compress(data,selector)

返回selector为True的data对应元素 例如：

importitertools

foreachinitertools.compress('abcd',[1,0,1,0]):

printeach

输出：

a

c

itertools.count(start=0,step=1)

返回以start开始，step递增的序列，无限递增 例如：

importitertools

foreachinitertools.count(start=0,step=2):

printeach

输出：

1

2

3

.

.

itertools.cycle(iterable)

将迭代器进行无限迭代 例如：

importitertools

foreachinitertools.cycle('ab'):

printeach

输出：

a

b

a

b

.

.

itertools.dropwhile(predicate, iterable)

直到predicate为真，就返回iterable后续数据， 否则drop掉 例如：

importitertools

foreachinitertools.dropwhile(lambdax:x<5,[2,1,6,8,2,1]):

printeach

输出：

6

8

2

1

itertools.groupby(iterable[,key])

返回一组(key,itera),key为iterable的值，itera为等于key的所有项 例如:

importitertools

forkey,valeinitertools.groupby('aabbbc'):

printkey,list(vale)

输出:

a['a','a']

b['b','b','b']

c['c']

itertools.ifilter(predicate, iterable)

返回predicate结果为True的元素迭代器，如果predicate为None,则返回所有iterable中为True的项 例如：

importitertools

forvalueinitertools.ifilter(lambdax:x%2,range(10)):

printvalue

输出:

1

3

5

7

9

itertools.ifilterfasle(predicate,iterable)

返回predicate为False的元素，如果predicate为None,则返回所有iterable中为False的项 例如：

importitertools

forvalueinitertools.ifilterfalse(lambdax:x%2,range(10)):

printvalue

输出:

0

2

4

6

8

itertools.imap(function,*iterables)

相当于迭代器方式的map() 例如：

importitertools

forvalueinitertools.imap(lambdax,y:xy,(1,2,3),(4,5,6)):

printvalue

输出:

5

7

9

itertools.islice(iterable, start,stop[,step])

相当于迭代器方式的切片操作 例如：

importitertools

forvalueinitertools.islice('abcdefg',1,4,2):

printvalue

输出:

b

d

itertools.repeat(object,[,times])

不停的返回object对象，如果指定了times,则返回times次 例如：

importitertools

forvalueinitertools.repeat('a',2):

printvalue

输出:

a

a

itertools.starmap(function,iterable)

返回function(iter)的值，iter为iterable的元素 例如：

importitertools

forvalueinitertools.starmap(lambdax,y:x*y,[(1,2),(3,4)]):

printvalue

输出:

2

12

itertools.takewhile(predicate,iterable)

如果predicate为真，则返回iterable元素，如果为假则不再返回，break. 例如：

importitertools

forvalueinitertools.takewhile(lambdax:x<5,[1,3,5,6]):

printvalue

输出:

1

3