Python标准库：迭代器Itertools

时间：2014-06-19 11:14:58 阅读：359 评论：0 收藏：0 [点我收藏+]

Infinite Iterators:

Iterator	Arguments	Results	Example
`count()`	start, [step]	start, start+step, start+2*step, ...	`count(10) --> 10 11 12 13 14 ...`
`cycle()`	p	p0, 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 times	`repeat(10, 3) --> 10 10 10`

Iterators terminating on the shortest input sequence:

Iterator	Arguments	Results	Example
`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, seq	seq[n], seq[n+1], starting when pred fails	`dropwhile(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, seq	elements of seq where pred(elem) is true	`ifilter(lambda x: x%2, range(10)) --> 1 3 5 7 9`
`ifilterfalse()`	pred, seq	elements of seq where pred(elem) is false	`ifilterfalse(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, seq	func(seq[0]), func(seq[1]), ...	`starmap(pow, [(2,5), (3,2), (10,3)]) --> 32 9 1000`
`tee()`	it, n	it1, it2, ... itn splits one iterator into n
`takewhile()`	pred, seq	seq[0], seq[1], until pred fails	`takewhile(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=‘-‘) --> Ax By C- D-`

Combinatoric generators:

Iterator	Arguments	Results
`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, r	r-length tuples, in sorted order, no repeated elements
`combinations_with_replacement()`	p, r	r-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(iterable, r)

从可迭代对象中选取r个对象的所有组合。

共有n! / r! / (n-r)!种组合，如果r大于了可迭代对象的个数，那么结果为空

itertools.combinations_with_replacement(iterable, r)

有放回的抽取组合。

共： (n+r-1)! / r! / (n-1)!

itertools.compress(data, selectors)

选择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=0, step=1)

创建一个无限的迭代器，从start开始，步长为step。

itertools.cycle(iterable)

创建一个循环的迭代器，以迭代器本身无限循环。

itertools.dropwhile(predicate, iterable)

创建一个迭代器，丢弃符合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(predicate, iterable)

与dropwhile相反的一个函数，它用来筛选中意的元素。

itertools.ifilterfalse(predicate, iterable)

与ifilter相反，选择不满足predicate条件的元素，与dropwhile类似。

itertools.imap(function, *iterables)

将迭代器送入第一个元素function，经过函数的加工后，输出到目标的迭代器。

后面的n个迭代器分别代表function中的n个argument。

itertools.izip(*iterables)

将多个迭代器进行组合，方法类似于zip，只是zip输出list。

itertools.islice(iterable, start, stop[, 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(predicate, iterable)

返回满足predicate条件的迭代器中的元素。与ifilter类似。

itertools.tee(iterable[, n=2])

从迭代器中创建n个相同的迭代器。

itertools.starmap(function, iterable)

starmap与imap有什么不同呢？starmap需要自己把送入function的argument打包，比如先用izip将两个迭代器打包，然后送入starmap的需要两个argument的function。

Python标准库：迭代器Itertools,布布扣,bubuko.com

Python标准库：迭代器Itertools

标签：python object 类迭代器 iterator

原文地址：http://blog.csdn.net/frog_in_a_well/article/details/30095953

踩

(0)

评论一句话评论（0）

分享档案

更多>

2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)

周排行