import itertools
from collections.abc import Iterator

'''
count(start, step)
    count會(huì)返回一個(gè)迭代器對(duì)象着憨，start的默認(rèn)值為0,step的默認(rèn)值為1妒御。返回的迭代器從start開始甘桑，步長(zhǎng)為step茬高。
可以無限迭代瘩欺。
'''
res = itertools.count()
print(isinstance(res, Iterator))        # True
for i in itertools.count(1, 2):
    if i > 100:
        break
    print(i)

'''
islice(iterable, stop):
    返回一個(gè)迭代器讨越，該迭代器只能迭代stop次想邦。
islice(iterable, start, stop[, step])：
    返回一個(gè)迭代器裤纹，該迭代器跳過start個(gè)元素以后開始迭代，總共迭代stop次后結(jié)束丧没，可以指定步長(zhǎng)鹰椒，stop指定了該迭代器
元素的總個(gè)數(shù)，被start跳過的和step跳過的都計(jì)算在內(nèi)呕童。
'''
res = itertools.islice(range(10), 5)
print(isinstance(res, Iterator))  # True

for i in itertools.islice(itertools.count(10), 5):
    print(i)
# 10 11 12 13 14
for i in itertools.islice(itertools.count(10), 5, 10):
    print(i)
# 15 16 17 18 19
for i in itertools.islice(itertools.count(10), 5, 10, 2):
    print(i)
# 15 17 19

'''
cycle(iterable):
    返回一個(gè)可以重復(fù)循環(huán)的迭代器,該迭代器一直在iterable里面循環(huán)
'''
res = itertools.cycle('abc')
print(isinstance(res, Iterator))
count = 0
for i in res:
    if count > 10:
        break
    count += 1
    print(i)
'''
accumulate(iterable[, func])
    返回一個(gè)迭代器漆际，該迭代中的結(jié)果是對(duì)iterable中的對(duì)象進(jìn)行逐一累加，最后將結(jié)果組成一個(gè)迭代器夺饲。
    比如：
        [1, 2, 3, 4]--->[1,3,6,10]
    或者接收一個(gè)操作函數(shù)奸汇，按照函數(shù)進(jìn)行操作
'''
def operator(x, y):
    return x + y


print(list(itertools.accumulate(range(10))))
# [0, 1, 3, 6, 10, 15, 21, 28, 36, 45]
print(list(itertools.accumulate(range(10), operator)))
# [1, 3, 6, 10, 15, 21, 28, 36, 45]

'''
chain(*iterable)
　　  chain能夠?qū)⒍鄠€(gè)可迭代對(duì)象合并成一個(gè)更長(zhǎng)的可迭代對(duì)象,返回一個(gè)迭代器對(duì)象。
'''
list1 = ['張三', '李四', '王五']
list2 = [18, 19, 20]
list3 = ['上海', '重慶', '成都']
res = itertools.chain(list1, list2, list3)
print(isinstance(res, Iterator))    # True
for i in res:
    print(i)
'''
combinations(iterable, r)
    從iterable中選擇r個(gè)不同的元素往声，返回所有的可能相當(dāng)實(shí)現(xiàn)C(m, n)算法擂找，結(jié)果為迭代器
'''
print(list(itertools.combinations(range(10), 5)))
'''
combinations_with_replacement(iterable, r) 
    允許重復(fù)元素的組合
'''
print(list(itertools.combinations_with_replacement('abcd', 3)))
'''
compress(data, selectors)
    按照真值表篩選元素，data為需要篩選的元素浩销，selectors為真值表
'''
print(list(itertools.compress('abcdefg', [1, 0, 1, 1, 0, 1, 1])))
'''
dropwhile(predicate, iterable)
    根據(jù)predicate定義的條件贯涎，過濾掉不滿足條件的元素
'''
print(list(itertools.dropwhile(lambda x: x < 5, range(10))))
'''
filterfalse(function or None, sequence)
    根據(jù)function中定義的條件，過濾掉滿足條件的元素慢洋。
'''
print(list(itertools.filterfalse(lambda x: x < 5, [1, 2, 3, 4, 5, 6, 7, 8, 9, 10])))
'''
groupby(iterable, key=None)
    按照key對(duì)元素進(jìn)行分組塘雳，返回值類似于字典陆盘，鍵為key的返回值，值為分組后的結(jié)果败明，如果key為None,那么鍵為被分組的值
本身隘马。
'''
x = itertools.groupby(range(10), key=lambda x: x > 8 or x < 5)
for condition, numbers in x:
    print(condition, list(numbers))
'''
permutations(iterable[, r])
    輸出輸入序列的全排列，如果沒有r參數(shù)妻顶，則對(duì)iterable進(jìn)行全排列酸员，如果有r參數(shù)，則先從iterable中選出r個(gè)不同的元素盈包，再
進(jìn)行全排列沸呐。
'''
print(list(itertools.permutations([1,2,3,4], 3)))
'''
product(*iterables, repeat=1) 
    返回笛卡爾積，repeat默認(rèn)為1
    product(A, B)相當(dāng)于((x, y) for x in A for y in B)
    product(A, repeat=2)相當(dāng)于product(A, A)
'''
print(list(itertools.product([1, 2, 3, 4],repeat=1)))
print(list(itertools.product([1, 2, 3, 4], ['a', 'b', 'c'])))
print(list(itertools.product([1, 2, 3, 4], ['a', 'b', 'c'], repeat=2)))
'''
repeat(object[, times ])
    返回重復(fù)元素呢燥，如果不傳入times參數(shù)崭添，將生成一個(gè)可以無窮無盡產(chǎn)生重復(fù)元素的迭代器
'''
result = itertools.repeat(10, 4)
print(isinstance(result, Iterator))     # True
for i in result:
    print(i)
print(list(map(pow, range(10), itertools.repeat(2))))
'''
starmap(function, sequence)
    將sequence中的所有元素都作用于function,將結(jié)果組合成一個(gè)迭代器
'''
print(list(itertools.starmap(str.islower, 'hfwehSJHWsfhws')))
'''
itertools.takewhile(predicate, iterable)
    將iterable的所有元素一次作用于predicate中，直到返回的結(jié)果為False為止叛氨。將結(jié)果為True的所有元素組成一個(gè)迭代器呼渣。
    takewhile(lambda x: x<5, [1,4,6,4,1]) --> 1 4
'''
print(list(itertools.takewhile(lambda x:x < 10, range(20))))
'''
itertools.tee(iterable, n=2)
    生成n個(gè)元素相同的獨(dú)立的迭代器，迭代器中的元素和iterable中的元素相同
'''
x = itertools.tee(range(5), 2)
for i in x:
    print(list(i))
'''
itertools.zip_longest(*iterables, fillvalue=None)
    zip_longest('ABCD', 'xy', fillvalue='-') --> Ax By C- D-
    按照最長(zhǎng)的iterable進(jìn)行組合寞埠，可以設(shè)定填充值屁置，默認(rèn)為None
'''
print(list(itertools.zip_longest(range(3), range(5))))
s = [('Python', 1), ('Swift', 2), ('Python', 3), ('Swift', 4), ('Python', 9)]
d = {}
for k, v in s:
    d.setdefault(k, []).append(v)       # setdefault方法是當(dāng)字典中查不到這個(gè)key對(duì)應(yīng)的值，就設(shè)定為默認(rèn)值
print(list(d.items()))