Python模块：itertools

itertools模块：循环器

 

一，无穷循环器：count,cycle,repeat

（1）count(5,3)   #从5开始的整数循环器，每次增加3，即：5,8,11,14,17...

from itertools import *
import time
a = count(5,3)
for i in a:
    print(i)
    time.sleep(1)

输出结果为：
5
8
11
14
17
20
23
26

（2）cycle('zxy')  #重复元素x y z x y z x y z... 

from itertools import *
import time

s = cycle('xyz')

for i in s:
    print(i)
    time.sleep(1)

输出结果为：
x
y
z
x
y
z
x
y
z

repeat()   #重复元素

例1：
from itertools import *
import time

s = repeat(3.14)  #无限重复元素

for i in s:
    print(i)
    time.sleep(1)

输出结果为：
3.14
3.14
3.14
3.14
3.14
3.14

例2：
from itertools import *
import time

s = repeat(3,5)   #重复元素3，共5次

for i in s:
    print(i)
    time.sleep(1)

输出结果为：
3
3
3
3
3

 

二，函数式工具：starmap,takewhile,dropwhile

（1）starmap()    #跟map类似

from itertools import *

s = starmap(pow,[(1,1),(2,2),(3,3)])  #pow()求指数1**1,2**2,3**3
for i in s:
    print(i)

输出结果为：
1
4
27

（2）takewhile()   #当函数返回True时，收集元素到循环器。一旦函数返回False，则停止。

from itertools import *

s1 = takewhile(lambda x: x < 5, [1,2,3,4,5,6,7])
for i in s1:
    print(i)

输出结果为：
1
2
3
4

（3）dropwhile()   #与takewhile相反。

s2 = dropwhile(lambda x: x < 5, [1,2,3,4,5,6,7])
for i in s2:
    print(i)

输出结果为：
5
6
7