import itertools
from collections.abc import Iterator

'''
count(start, step)
    count会返回一个迭代器对象，start的默认值为0,step的默认值为1。返回的迭代器从start开始，步长为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):
    返回一个迭代器，该迭代器只能迭代stop次。
islice(iterable, start, stop[, step])：
    返回一个迭代器，该迭代器跳过start个元素以后开始迭代，总共迭代stop次后结束，可以指定步长，stop指定了该迭代器
元素的总个数，被start跳过的和step跳过的都计算在内。
'''
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):
    返回一个可以重复循环的迭代器,该迭代器一直在iterable里面循环
'''
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])
    返回一个迭代器，该迭代中的结果是对iterable中的对象进行逐一累加，最后将结果组成一个迭代器。
    比如：
        [1, 2, 3, 4]--->[1,3,6,10]
    或者接收一个操作函数，按照函数进行操作
'''
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能够将多个可迭代对象合并成一个更长的可迭代对象,返回一个迭代器对象。
'''
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个不同的元素，返回所有的可能相当实现C(m, n)算法，结果为迭代器
'''
print(list(itertools.combinations(range(10), 5)))
'''
combinations_with_replacement(iterable, r) 
    允许重复元素的组合
'''
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)
    根据predicate定义的条件，过滤掉不满足条件的元素
'''
print(list(itertools.dropwhile(lambda x: x < 5, range(10))))
'''
filterfalse(function or None, sequence)
    根据function中定义的条件，过滤掉满足条件的元素。
'''
print(list(itertools.filterfalse(lambda x: x < 5, [1, 2, 3, 4, 5, 6, 7, 8, 9, 10])))
'''
groupby(iterable, key=None)
    按照key对元素进行分组，返回值类似于字典，键为key的返回值，值为分组后的结果，如果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参数，则对iterable进行全排列，如果有r参数，则先从iterable中选出r个不同的元素，再
进行全排列。
'''
print(list(itertools.permutations([1,2,3,4], 3)))
'''
product(*iterables, repeat=1) 
    返回笛卡尔积，repeat默认为1
    product(A, B)相当于((x, y) for x in A for y in B)
    product(A, repeat=2)相当于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 ])
    返回重复元素，如果不传入times参数，将生成一个可以无穷无尽产生重复元素的迭代器
'''
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,将结果组合成一个迭代器
'''
print(list(itertools.starmap(str.islower, 'hfwehSJHWsfhws')))
'''
itertools.takewhile(predicate, iterable)
    将iterable的所有元素一次作用于predicate中，直到返回的结果为False为止。将结果为True的所有元素组成一个迭代器。
    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个元素相同的独立的迭代器，迭代器中的元素和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-
    按照最长的iterable进行组合，可以设定填充值，默认为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方法是当字典中查不到这个key对应的值，就设定为默认值
print(list(d.items()))