昨天见到了一个python的内建模块-itertools。今天总结一下:
itertools提供了非常有用的用于操作迭代对象的函数,整理如下
首先,我们看看itertools提供的几个“无限”迭代器
- count()
eg:
import itertools
arr = itertools.count(1)
for n in arr:
print(n)
这里会从1开始打印出序列,程序不会结束,除非自己中断程序。如果上面count()传2,就会从2开始递增打印。如下:
2. cycle()
eg::
import itertools
s = itertools.cycle('ABC')
for c in s:
print(c)
这个方法会把传入的一个序列无限重复下去,如下:
3. repeat()
eg:
import itertools
s = itertools.repeat('A', 5)
for n in s:
print(n)
负责把一个元素无限重复下去,提供的第二个参数就可以限定重复次数,如下:
注意:
无限序列虽然可以无限迭代下去,但是通常会通过takewhile()等函数根据条件判断来截取出一个有限的序列
eg:
import itertools
arr = itertools.count(1)
s = itertools.takewhile(lambda x: x <= 10, arr)
for n in s:
print(n)
这样就会打印出1到10,不会无限迭代。
了解itertools提供的几个迭代器操作函数
- chain()
eg:
import itertools
for c in itertools.chain('ABC', 'XYZ'):
print(c)
chain()可以把一组迭代对象串联起来,形成一个更大的迭代器,看运行结果:
2. groupby()
eg:
import itertools
for key, group in itertools.groupby('AAABBBCCAAA'):
print(key, list(group))
把迭代器中相邻的重复元素挑出来放在一起,
说明:
这里的挑选规则是通过函数完成的,只要作用于函数的两个元素返回的值相等,这两个元素就被认为是在一组的,而函数返回值作为组的key。
如果我们要忽略大小写分组,就可以让元素’A’和’a’都返回相同的key,将小写转化成大写:
eg:
import itertools
for key, group in itertools.groupby('AaaBBbcCAAa', lambda c: c.upper()):
print(key, list(group))