python-itertools

昨天见到了一个python的内建模块-itertools。今天总结一下：

itertools提供了非常有用的用于操作迭代对象的函数，整理如下

首先，我们看看itertools提供的几个“无限”迭代器

1. 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提供的几个迭代器操作函数

1. 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。

import itertools
for key, group in itertools.groupby('AaaBBbcCAAa', lambda c: c.upper()):
   print(key, list(group))