今天学了简单好玩的模块。itertools模块，简单的说，itertools模块就是产生一个循环器

来看下这个模块的功能都有哪些吧

无穷循环器

count()  从開始的数字一直数下去

count(10)   #-->  10 11 12 13 14 15 …

count(10,2) #-->  10 12 14 16 18 20 …

cycle()  不停的将可迭代元素进行循环

cycle(‘abcd’)  #-->  a b c d a b c d a …

repeat()  不停的反复一个元素

repeat(20)   #-->  20 20 20 20 20 …

repeat(20,3) #-->  20 20 20

有限循环器

accumulate()  将可迭代的容器进行累加，也能够自己指定详细的操作

accumulate(range(10))    #-->  0 1 3 6 10

accumulate(range(1,5), func = operator.mul)    #-->  1 2 6 24

chain()  将几个可迭代的容器依次迭代

chain(range(5),'abcd')          #-->  0 1 2 3 4 a b c d

chain(['abcd', 'bcde', 'cdef'])    #-->  abcd bcde cdef

chain.from_iterable()  将可迭代容器里面的元素再次进行迭代一次

chain.from_iterable(['abcd', 'bcde', 'cdef'])            #-->  a b c d b c d e c d e f

chain.from_iterable([['abcd', 'bcde','cdef'],'abcd'])    #-->  abcd bcdecdef a b c d

compress()  两个參数分别为data, selectors, 依据selectors中的真假情况返回data中的元素

compress('ABCDEF', [1,0,1,0,1,1])       #-->  A C E F

dropwhile()  原型为dropwhile(predicate, iterable)当predicate返回True时，跳过元素。一旦函数返回False，则開始收集剩下的全部元素到循环器

dropwhile(lambda x: x<5, [1,4,6,4,1])    #-->  6 4 1

filterfalse()  原型为filterfalse(predicate, iterable)当predicate返回False时，才将iterable中的元素加入进循环器

filterfalse(lambda x : x%2 , range(10))    #-->  0 2 4 6 8

groupby()  原型为groupby(iterable, key=None)将key的结果作用于iterable中的元素，将拥有同样返回结果的元素增加到循环器中。该函数之前须要确保iterable是经过排序的

演示样例一：

for i,g in itertools.groupby('AAAABBBCCDAABBB'):

print(i,list(g))

结果：

A ['A', 'A', 'A','A']

B ['B', 'B', 'B']

C ['C', 'C']

D ['D']

A ['A', 'A']

B ['B', 'B', 'B']

演示样例二：

def height_classify(h):

if h>180 :

return 'tall'

elif h< 160 :

return 'short'

else :

return 'middle'

friends = [192, 158, 168, 195, 185, 170, 135,174, 182]

friends = sorted(friends, key =height_classify)

for m, n in itertools.groupby(friends, key =height_classify):

print(m)

print(list(n))

结果：

middle

[168, 170, 174]

short

[158, 135]

tall

[192, 195, 185, 182]

islice() 原型为islice(iterable, stop)或islice(iterable, start, stop[, step])

islice('ABCDEFG', 2)             #-->  A B

islice('ABCDEFG', 2, 4)          #-->  C D

islice('ABCDEFG', 2, None)      #-->  C D E F G

islice('ABCDEFG', 0, None, 2)   #-->  A C E G

starmap() 原型为starmap(function, iterable)

starmap(pow, [(2,5), (3,2), (10,3)])      #-->  32 9 1000

takewhile() 原型为takewhile(predicate,iterable)，与dropwhile()功能相反

takewhile(lambda x: x<5, [1,4,6,4,1])    #--> 1 4

tee() 原型为tee(iterable, n=2); 从单个的iterable返回n个独立的循环器

zip_longest() 原型为zip_longest(*iterables,fillvalue=None)

zip_longest('abcd','123',fillvalue='*')      #-->  ('a', '1')('b','2')('c', '3')('d', '*')

组合生成器

product() 原型为product(*iterables, repeat=1)做笛卡尔乘积

product('abc','xy')    #-->  ('a', 'x')('a', 'y')('b', 'x')('b', 'y')('c','x')('c', 'y')

permutations() 原型为permutations(iterable,r=None)全排列

permutations('abc')    #-->  ('a', 'b', 'c')('a', 'c', 'b')('b', 'a', 'c')('b','c', 'a')('c', 'a', 'b')('c', 'b', 'a')

permutations('abcd', r=2)    #-->  ('a', 'b')('a', 'c')('a', 'd')('b', 'a')('b','c')('b', 'd')('c', 'a')('c', 'b')('c', 'd')('d', 'a')('d', 'b')('d', 'c')

combinations() 原型为combinations(iterable, r)

combinations('abcd',r=3)    #-->  ('a', 'b', 'c')('a', 'b', 'd')('a', 'c', 'd')('b','c', 'd')

combinations_with_replacement() 原型为combinations_with_replacement(iterable,r)

combinations_with_replacement('abc',r=3)    #-->  ('a', 'a', 'a')('a', 'a', 'b')('a', 'a', 'c')('a','b', 'b')('a', 'b', 'c')('a', 'c', 'c')('b', 'b', 'b')('b', 'b', 'c')('b', 'c','c')('c', 'c', 'c')