import itertools

c = itertools.chain([1, 2, 3], "abc", {"k1": "v1", "k2": "v2"})
# 直接打印的話是一個(gè)對(duì)象
print(c) 
"""
<itertools.chain object at 0x00000000029745F8>
"""
print(list(c)) 
"""
1 2 3 a b c k1 k2
"""

# 還可以使用 chain.from_iterable
# 參數(shù)接收多個(gè)可迭代對(duì)象組成的一個(gè)可迭代對(duì)象
c = itertools.chain.from_iterable(
    [[1, 2, 3], "abc", {"k1": "v1", "k2": "v2"}]
)
print(list(c)) 
"""
1 2 3 a b c k1 k2
"""

import itertools

# 內(nèi)置的 zip 是把多個(gè)迭代器對(duì)象中的每一個(gè)元素按照順序組合到一個(gè)元組中
name = ["高老師", "豬哥", "S 佬"]
where = ["江蘇", "北京", "深圳"]
z = zip(name, where)
print(z)
"""
<zip object at 0x00000257F3FEBEC0>
"""
print(list(z))
"""
[('高老師', '江蘇'), ('豬哥', '北京'), ('S 佬', '深圳')]
"""

# 但如果兩者長(zhǎng)度不一致怎么辦？
name = ["高老師", "豬哥", "S 佬", "xxx"]
where = ["江蘇", "北京", "深圳"]
print(list(zip(name, where)))
"""
[('高老師', '江蘇'), ('豬哥', '北京'), ('S 佬', '深圳')]
"""
# 可以看到，長(zhǎng)度不一致的時(shí)候，當(dāng)一方結(jié)束之后就停止匹配

# 如果想匹配長(zhǎng)的，那么可以使用 itertools 下面的 zip_longest
print(list(itertools.zip_longest(name, where))) 
"""
[('高老師', '江蘇'), ('豬哥', '北京'), ('S 佬', '深圳'), ('xxx', None)]
"""
# 默認(rèn)使用 None 進(jìn)行匹配，當(dāng)然我們也可以指定內(nèi)容
print(list(itertools.zip_longest(name, where, fillvalue="中國")))
"""
[('高老師', '江蘇'), ('豬哥', '北京'), ('S 佬', '深圳'), ('xxx', '中國')]
"""

import itertools

num = range(20)
# 選擇 index=5 到 index=10（不包含）的位置
s = itertools.islice(num, 5, 10)
print(list(s))  # [5, 6, 7, 8, 9]

# 選擇開頭到 index=5 的位置
s = itertools.islice(num, 5)
print(list(s))  # [0, 1, 2, 3, 4]

# 選擇從 index=5 到 index=15（不包含）的位置，步長(zhǎng)為 3
s = itertools.islice(num, 5, 15, 3)
print(list(s))  # [5, 8, 11, 14]

import itertools

r = [1, 2, 3, 4, 5]
i1, i2 = itertools.tee(r, 2)
print(list(i1))  # [1, 2, 3, 4, 5]
print(list(i2))  # [1, 2, 3, 4, 5]

import itertools

"""
count(start=0, step=1) 返回一個(gè)迭代器，負(fù)責(zé)無限地生成連續(xù)的整數(shù)
接收兩個(gè)參數(shù)：起始（默認(rèn)為0）和步長(zhǎng)（默認(rèn)為1）
等價(jià)于：
def count(firstval=0, step=1):
    x = firstval
    while 1:
        yield x
        x += step
"""
# 起始值為 5，步長(zhǎng)為 2
c1 = itertools.count(5, 2)
print(list(itertools.islice(c1, 5))) 
"""
[5, 7, 9, 11, 13]
"""

import itertools

"""
cycle(iterable) 返回一個(gè)迭代器，會(huì)無限重復(fù)里面的內(nèi)容，直到內(nèi)存耗盡
"""
c2 = itertools.cycle("abc")
print(list(itertools.islice(c2, 10)))
"""
['a', 'b', 'c', 'a', 'b', 'c', 'a', 'b', 'c', 'a']
"""

import itertools

"""
repeat(obj, times=None)，無限重復(fù) obj，除非指定 times
"""
# 重復(fù)指定的次數(shù)
print(list(itertools.repeat("abc", 3))) 
"""
['abc', 'abc', 'abc']
"""

import itertools

l = [1, 2, 3, 4, 5]
drop_l = itertools.dropwhile(lambda x: x < 3, l)
# 依舊返回迭代器
print(drop_l) 
"""
<itertools.dropwhile object at 0x000001AD63AD0488>
"""
# 可以看到小于3的都被丟掉了
print(list(drop_l))  
"""
[3, 4, 5]
"""

import itertools

l = [1, 2, 3, 4, 5]
take_l = itertools.takewhile(lambda x: x < 3, l)
print(take_l) 
"""
<itertools.takewhile object at 0x000001D37F512948>
"""
print(list(take_l)) 
"""
[1, 2]
"""
filter_l = filter(lambda x: x < 3, l)
print(list(filter_l))  
"""
[1, 2]
"""

import itertools

condition = [True, False, True, True, False]
data = [1, 2, 3, 4, 5]
print(list(itertools.compress(data, condition))) 
"""
[1, 3, 4]
"""

# 除了指定 True 和 False，還可以使用 Python 其它類型的值
# 會(huì)以其對(duì)應(yīng)的布爾值作為判斷依據(jù)
condition = [1, 0, "x", "x", {}]  
print(list(itertools.compress(data, condition))) 
"""
[1, 3, 4]
"""

import itertools

print(list(itertools.accumulate(range(5))))  
"""
[0, 1, 3, 6, 10]
"""
print(list(itertools.accumulate("abcde")))  
"""
["a", "ab", "abc", "abcd", "abcde"]
"""
# 所以這里的相加還要看具體的含義
try:
    print(list(itertools.accumulate([[1, 2], (3, 4)])))
except TypeError as e:
    print(e)  
    """
    can only concatenate list (not "tuple") to list
    """
    # 這里就顯示無法將列表和元組相加

# 當(dāng)然也可以自定義
data = [1, 2, 3, 4, 5]
method = lambda x, y: x * y
print(list(itertools.accumulate(data, method))) 
"""
[1, 2, 6, 24, 120]
"""
# 可以看到這里的結(jié)果就改變了

import itertools

print(list(itertools.product([1, 2, 3], [2, 3]))) 
"""
[(1, 2), (1, 3), (2, 2), (2, 3), (3, 2), (3, 3)]
"""

import itertools

data = [1, 2, 3, 4]
print(list(itertools.permutations(data)))
# 根據(jù)排列組合，顯然是 A44，總共 4 * 3 * 2 * 1 = 24 種組合
"""
[(1, 2, 3, 4), (1, 2, 4, 3), (1, 3, 2, 4), (1, 3, 4, 2), (1, 4, 2, 3), (1, 4, 3, 2),
(2, 1, 3, 4), (2, 1, 4, 3), (2, 3, 1, 4), (2, 3, 4, 1), (2, 4, 1, 3), (2, 4, 3, 1),
(3, 1, 2, 4), (3, 1, 4, 2), (3, 2, 1, 4), (3, 2, 4, 1), (3, 4, 1, 2), (3, 4, 2, 1),
(4, 1, 2, 3), (4, 1, 3, 2), (4, 2, 1, 3), (4, 2, 3, 1), (4, 3, 1, 2), (4, 3, 2, 1)]
"""

# 結(jié)果是 A42，總共 4 * 3 = 12 種組合
print(list(itertools.permutations(data, 2)))
"""
[(1, 2), (1, 3), (1, 4), 
 (2, 1), (2, 3), (2, 4), 
 (3, 1), (3, 2), (3, 4), 
 (4, 1), (4, 2), (4, 3)]
"""

import itertools

# permutations 只要順序不同就看做一種結(jié)果
# combinations 則保證只要元素相同就是同一種結(jié)果
data = "abcd"
print(list(itertools.combinations(data, 3)))  
"""
[('a', 'b', 'c'), ('a', 'b', 'd'), ('a', 'c', 'd'), ('b', 'c', 'd')]
"""
# 如果拿抽小球來作比喻的話，顯然 combinations 是不放回的，也就是不會(huì)重復(fù)單個(gè)的輸入元素
# 但有時(shí)候可能也需要考慮包含重復(fù)元素的組合，相當(dāng)于抽小球的時(shí)候有放回
# 對(duì)于這種情況，可以使用 combinations_with_replacement
print(list(itertools.combinations_with_replacement(data, 3)))
"""
[('a', 'a', 'a'), ('a', 'a', 'b'), ('a', 'a', 'c'), ('a', 'a', 'd'), ('a', 'b', 'b'),
('a', 'b', 'c'), ('a', 'b', 'd'), ('a', 'c', 'c'), ('a', 'c', 'd'), ('a', 'd', 'd'),
('b', 'b', 'b'), ('b', 'b', 'c'), ('b', 'b', 'd'), ('b', 'c', 'c'), ('b', 'c', 'd'),
('b', 'd', 'd'), ('c', 'c', 'c'), ('c', 'c', 'd'), ('c', 'd', 'd'), ('d', 'd', 'd')]
"""