Python编程：itertools模块

import itertools

# “无限”迭代器

# 创建一个无限的迭代器，代码会打印出自然数序列，根本停不下来，只能按Ctrl+C退出。
def testCount(): natuals=itertools.count(1) for i in natuals: print(i)
# testCount()


# 传入的一个序列无限重复下去
def testCycle(): cs=itertools.cycle("ABC") for i in cs: print(i)
# testCycle()

# 责把一个元素无限重复下去，不过如果提供第二个参数就可以限定重复次数
def testRepeat(): r=itertools.repeat("A",10) for n in r: print(n)

# testRepeat()

# 打印出1到10
# 通过takewhile()等函数根据条件判断来截取出一个有限的序列
def testTakewhile(): natuals=itertools.count(1)#从1开始 ns=itertools.takewhile(lambda x:x<=10, natuals) for n in ns: print(n)

# testTakewhile()

# chain()可以把一组迭代对象串联起来，形成一个更大的迭代器
# 迭代效果：'A' 'B' 'C' 'X' 'Y' 'Z'
def testchain(): cs=itertools.chain("ABC","XYZ") for c in cs: print(c)
# testchain()

# 把迭代器中相邻的重复元素挑出来放在一起
def testGroupby(): for key,group in itertools.groupby("AAAaaABBBCCCDDD"): print(key,list(group))
# testGroupby()
"""
A ['A', 'A', 'A']
B ['B', 'B', 'B']
C ['C', 'C']
A ['A', 'A', 'A']
"""

# 忽略大小写分组
def testGroupbyupper(): for key,group in itertools.groupby("AAaAABBbCCCddDDD",lambda x:x.upper()): print(key,list(group))
# testGroupbyupper()

# imap()和map()的区别在于，imap()可以作用于无穷序列，并且，如果两个序列的长度不一致，以短的那个为准。
def testimap(): for i in map(lambda x,y:x*y,[10,20,30],itertools.count(1)): print(i)

testimap()
  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
  
21
  
22
  
23
  
24
  
25
  
26
  
27
  
28
  
29
  
30
  
31
  
32
  
33
  
34
  
35
  
36
  
37
  
38
  
39
  
40
  
41
  
42
  
43
  
44
  
45
  
46
  
47
  
48
  
49
  
50
  
51
  
52
  
53
  
54
  
55
  
56
  
57
  
58
  
59
  
60
  
61
  
62
  
63
  
64
  
65
  
66
  
67
  
68
  
69
 

参考：
廖雪峰python-itertools

文章来源: pengshiyu.blog.csdn.net，作者：彭世瑜，版权归原作者所有，如需转载，请联系作者。

原文链接：pengshiyu.blog.csdn.net/article/details/79172093