首页 > 代码库 > Python标准库:迭代器Itertools

Python标准库:迭代器Itertools

Infinite Iterators:

IteratorArgumentsResultsExample
count()start, [step]start, start+step, start+2*step, ...count(10) --> 10 11 12 13 14 ...
cycle()pp0, p1, ... plast, p0, p1, ...cycle(‘ABCD‘) --> A B C D A B C D ...
repeat()elem [,n]elem, elem, elem, ... endlessly or up to n timesrepeat(10, 3) --> 10 10 10


Iterators terminating on the shortest input sequence:

IteratorArgumentsResultsExample
chain()p, q, ...p0, p1, ... plast, q0, q1, ...chain(‘ABC‘, ‘DEF‘) --> A B C D E F
compress()data, selectors(d[0] if s[0]), (d[1] if s[1]), ...compress(‘ABCDEF‘, [1,0,1,0,1,1]) --> A C E F
dropwhile()pred, seqseq[n], seq[n+1], starting when pred failsdropwhile(lambda x: x<5, [1,4,6,4,1]) --> 6 4 1
groupby()iterable[, keyfunc]sub-iterators grouped by value of keyfunc(v) 
ifilter()pred, seqelements of seq where pred(elem) is trueifilter(lambda x: x%2, range(10)) --> 1 3 5 7 9
ifilterfalse()pred, seqelements of seq where pred(elem) is falseifilterfalse(lambda x: x%2, range(10)) --> 0 2 4 6 8
islice()seq, [start,] stop [, step]elements from seq[start:stop:step]islice(‘ABCDEFG‘, 2, None) --> C D E F G
imap()func, p, q, ...func(p0, q0), func(p1, q1), ...imap(pow, (2,3,10), (5,2,3)) --> 32 9 1000
starmap()func, seqfunc(*seq[0]), func(*seq[1]), ...starmap(pow, [(2,5), (3,2), (10,3)]) --> 32 9 1000
tee()it, nit1, it2, ... itn splits one iterator into n 
takewhile()pred, seqseq[0], seq[1], until pred failstakewhile(lambda x: x<5, [1,4,6,4,1]) --> 1 4
izip()p, q, ...(p[0], q[0]), (p[1], q[1]), ...izip(‘ABCD‘, ‘xy‘) --> Ax By
izip_longest()p, q, ...(p[0], q[0]), (p[1], q[1]), ...izip_longest(‘ABCD‘, ‘xy‘, fillvalue=http://www.mamicode.com/‘-‘) --> Ax By C- D-

Combinatoric generators:

IteratorArgumentsResults
product()p, q, ... [repeat=1]cartesian product, equivalent to a nested for-loop
permutations()p[, r]r-length tuples, all possible orderings, no repeated elements
combinations()p, rr-length tuples, in sorted order, no repeated elements
combinations_with_replacement()p, rr-length tuples, in sorted order, with repeated elements
product(‘ABCD‘, repeat=2) AA AB AC AD BA BB BC BD CA CB CC CD DA DB DC DD
permutations(‘ABCD‘, 2) AB AC AD BA BC BD CA CB CD DA DB DC
combinations(‘ABCD‘, 2) AB AC AD BC BD CD
combinations_with_replacement(‘ABCD‘, 2) AA AB AC AD BB BC BD CC CD DD


itertools.chain(*iterables)
将括号内的所有可迭代对象,共同顺序输出至一个迭代对象。

itertools.combinations(iterabler)
从可迭代对象中选取r个对象的所有组合。
共有n! / r! / (n-r)!种组合,如果r大于了可迭代对象的个数,那么结果为空

itertools.combinations_with_replacement(iterabler)
有放回的抽取组合。
共: (n+r-1)! / r! / (n-1)!

itertools.compress(dataselectors)
选择data中的元素来创建迭代器,selector相当于mask。
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)

itertools.count(start=0step=1)
创建一个无限的迭代器,从start开始,步长为step。

itertools.cycle(iterable)
创建一个循环的迭代器,以迭代器本身无限循环。

itertools.dropwhile(predicateiterable)
创建一个迭代器,丢弃符合predicate描述的那些元素。

itertools.groupby(iterable[key])
这个函数功能类似于SQL的分组。使用groupby前,首先需要使用相同的keyfunc对iterable进行排序,比如调用内建的sorted函数。然后,groupby返回迭代器,每次迭代的元素是元组(key值, iterable中具有相同key值的元素的集合的子迭代器)。
  • groupby([0, 0, 0, 1, 1, 1, 2, 2, 2]) --> (0, (0 0 0)) (1, (1 1 1)) (2, (2 2 2))
itertools.ifilter(predicateiterable)
与dropwhile相反的一个函数,它用来筛选中意的元素。

itertools.ifilterfalse(predicateiterable)
与ifilter相反,选择不满足predicate条件的元素,与dropwhile类似。

itertools.imap(function*iterables)
将迭代器送入第一个元素function,经过函数的加工后,输出到目标的迭代器。
后面的n个迭代器分别代表function中的n个argument。

itertools.izip(*iterables)
将多个迭代器进行组合,方法类似于zip,只是zip输出list。

itertools.islice(iterablestartstop[step])
返回一个迭代器,是目标迭代器的一个slice。这个方法如果后面只加两个参数,那么第二个参数是slice的结束。

itertools.izip_longest(*iterables[fillvalue])
以最长的的迭代器为准来组合迭代器,短的迭代器会被填充fillvalue值。

itertools.permutations(iterable[r])
返回迭代器中元素r个元素的所有组合,一共有:n! / (n-r)!种方式。

itertools.product(*iterables[repeat])
列举出多个迭代器的互相组合,一个有n1*n2*n3*......nn种组合方式,(在每个迭代器中取一个元素)

itertools.repeat(object[times])
将一个迭代器重复times次。
主要的用途是为imap和izip提供恒定值。

itertools.takewhile(predicateiterable)
返回满足predicate条件的迭代器中的元素。与ifilter类似。

itertools.tee(iterable[n=2])
从迭代器中创建n个相同的迭代器。

itertools.starmap(functioniterable)
starmap与imap有什么不同呢?starmap需要自己把送入function的argument打包,比如先用izip将两个迭代器打包,然后送入starmap的需要两个argument的function。