# Python的內建模組itertools提供了非常有用的用於操作迭代物件的函式
# 1、count()
import itertools
natuals = itertools.count(1)
# for n in natuals:
#      print(n)
# 輸出結果:  1,2,3.....n 的自然數，一直執行下去


# 通過takewhile()等函式根據條件判斷來截取出一個有限的序列
natuals2 = itertools.count(1)
ns = itertools.takewhile(lambda x: x <= 10, natuals2)
print(list(ns))
# 輸出結果: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]


# 2、cycle() 會把傳入的一個序列無限重複下去：
import itertools
cs = itertools.cycle('ABC')  # 注意字串也是序列的一種
# for c in cs:
#      print(c)
# 輸出結果:  'A', 'B', 'C' 無限重複， 一直輸出

# 3、repeat()負責把一個元素無限重複下去，不過如果提供第二個引數就可以限定重複次數
ns = itertools.repeat('A', 3)
for n in ns:
    print(n)

# 4、chain()可以把一組迭代物件串聯起來，形成一個更大的迭代器
for c in itertools.chain('ABC', 'XYZ'):
     print(c)
# 迭代效果：'A' 'B' 'C' 'X' 'Y' 'Z'


# 5、groupby()把迭代器中相鄰的重複元素挑出來放在一起
ls = []
for key, group in itertools.groupby('AAABBBCCAAA'):
    a = list(group)
    # print(type(group))   # 注意: 迭代器是迴圈完成之後，裡面就是空的，再次迴圈是會報錯的
    print(key, a)
    ls.append(a)
print(ls)
"""
輸出結果:
    A ['A', 'A', 'A']
    B ['B', 'B', 'B']
    C ['C', 'C']
    A ['A', 'A', 'A']
    [['A', 'A', 'A'], ['B', 'B', 'B'], ['C', 'C'], ['A', 'A', 'A']]
"""

# groupby()  裡面是可以寫匿名函式
for key, group in itertools.groupby('AaaBBbcCAAa', lambda x: x.upper()):
     print(key, list(group))

"""
輸出結果:
    A ['A', 'A', 'A']
    B ['B', 'B', 'B']
    C ['C', 'C']
    A ['A', 'A', 'A']
"""