1.count
itertools.count函数返回的生成器能生成多个数,如果不传入参数,itertools.count函数会生成从零开始的整数数列。不过,我们可以提供可选的start和step值。
import itertools
gen = itertools.count(1, .5)
next(gen) #1
next(gen) # 1.5
next(gen) # 2.0
.
.
.
然后itertools.count 从不停止。如果调用list(count())就会报内存泄露的错误。不过.itertools.takewhile函数则不同,他会生成一个使用另一个生成器的生成器。在指定的条件计算结果为False时停止。因此,可以把这两个函数结合在一起使用。
gen = itertools.takewhile(lambda n: n<3, itertools.count(1, .5))
list(gen) #[1, 1.5, 2.0, 2.5]
>>> def vowel(c):
... return c.lower() in 'aeiou'
...
>>> list(filter(vowel, 'Aardvark'))
['A', 'a', 'a']
>>> import itertools
>>> list(itertools.filterfalse(vowel, 'Aardvark'))
['r', 'd', 'v', 'r', 'k']
>>> list(itertools.dropwhile(vowel, 'Aardvark'))
['r', 'd', 'v', 'a', 'r', 'k']
>>> list(itertools.takewhile(vowel, 'Aardvark'))
['A', 'a']
>>> list(itertools.compress('Aardvark', (1,0,1,1,0,1)))
['A', 'r', 'd', 'a']
>>> list(itertools.islice('Aardvark', 4))
['A', 'a', 'r', 'd']
>>> list(itertools.islice('Aardvark', 4, 7))
['v', 'a', 'r']
>>> list(itertools.islice('Aardvark', 1, 7, 2))
['a', 'd', 'a']
>>> sample = [5, 4, 2, 8, 7, 6, 3, 0, 9, 1]
>>> import itertools
>>> list(itertools.accumulate(sample))
[5, 9, 11, 19, 26, 32, 35, 35, 44, 45]
>>> list(itertools.accumulate(sample, min))
[5, 4, 2, 2, 2, 2, 2, 0, 0, 0]
>>> list(itertools.accumulate(sample, max))
[5, 5, 5, 8, 8, 8, 8, 8, 9, 9]
>>> import operator
>>> list(itertools.accumulate(sample, operator.mul))
[5, 20, 40, 320, 2240, 13440, 40320, 0, 0, 0]
>>> list(itertools.accumulate(range(1, 11), operator.mul))
[1, 2, 6, 24, 120, 720, 5040, 40320, 362880, 3628800]
2. chain(*iterables)
创建一个迭代器,该迭代器从第一个迭代中返回元素,直到耗尽,然后继续进行下一个迭代,直到所有的迭代器都被耗尽.用于将连续序列作为单个序列进行处理,大致相当于:
def chain(*iterables):
# chain('ABC', 'DEF') --> A B C D E F
for it in iterables:
for element in it:
yield element
chain method.png
3.cycle(iterable)
让迭代器从迭代中返回元素,并保存每个元素的副本.当迭代耗尽时,从保存的副本返回元素.重复下去.大致相当于:
def cycle(iterable):
# cycle('ABCD') --> A B C D 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
cycle method.png
网友评论